diff options
Diffstat (limited to 'test/test_results.py')
| -rw-r--r-- | test/test_results.py | 14 |
1 files changed, 8 insertions, 6 deletions
diff --git a/test/test_results.py b/test/test_results.py index e7fcabf..0865686 100644 --- a/test/test_results.py +++ b/test/test_results.py @@ -310,10 +310,13 @@ class TestExtractorResults(unittest.TestCase): elif isinstance(test, range): self.assertRange(value, test, msg=path) elif isinstance(test, set): - try: - self.assertIn(value, test, msg=path) - except AssertionError: - self.assertIn(type(value), test, msg=path) + for item in test: + if isinstance(item, type) and isinstance(value, item) or \ + value == item: + break + else: + v = type(value) if len(str(value)) > 64 else value + self.fail(f"{v!r} not in {test}: {path}") elif isinstance(test, list): subtest = False for idx, item in enumerate(test): @@ -423,8 +426,7 @@ class ResultJob(job.DownloadJob): def run(self): self._init() - for msg in self.extractor: - self.dispatch(msg) + self.dispatch(self.extractor) def handle_url(self, url, kwdict, fallback=None): self._update_url(url) |