From d12a04af3c1fd6ce4b98e006cf64c5e5fc5e56fb Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Mike=20F=C3=A4hrmann?= Date: Sat, 17 Dec 2016 13:23:29 +0100 Subject: [PATCH] [test] filter out classes without a 'test' attribute --- gallery_dl/extractor/test.py | 6 +++++- 1 file changed, 5 insertions(+), 1 deletion(-) diff --git a/gallery_dl/extractor/test.py b/gallery_dl/extractor/test.py index 48b5333c..a23c87da 100644 --- a/gallery_dl/extractor/test.py +++ b/gallery_dl/extractor/test.py @@ -59,7 +59,11 @@ class TestExtractor(Extractor): if len(klass.test) > index ] else: - tests = [test for klass in klasses for test in klass.test] + tests = [ + test + for klass in klasses if hasattr(klass, "test") + for test in klass.test + ] if not tests: raise exception.NotFoundError("test")