New upstream version 1.11.1upstream/1.11.1

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():