diff options
Diffstat (limited to 'test/test_extractor.py')
| -rw-r--r-- | test/test_extractor.py | 16 |
1 files changed, 2 insertions, 14 deletions
diff --git a/test/test_extractor.py b/test/test_extractor.py index fa0709b..2555b58 100644 --- a/test/test_extractor.py +++ b/test/test_extractor.py @@ -157,24 +157,12 @@ class TestExtractor(unittest.TestCase): def capitalize(c): if "-" in c: return string.capwords(c.replace("-", " ")).replace(" ", "") - if "." in c: - c = c.replace(".", "") return c.capitalize() - mapping = { - "2chan" : "futaba", - "3dbooru": "threedeebooru", - "4chan" : "fourchan", - "4plebs" : "fourplebs", - "8chan" : "infinitychan", - "oauth" : None, - } - for extr in extractor.extractors(): - category = mapping.get(extr.category, extr.category) - if category: + if extr.category not in ("", "oauth"): expected = "{}{}Extractor".format( - capitalize(category), + capitalize(extr.category), capitalize(extr.subcategory), ) if expected[0].isdigit(): |