Fix test skipping

LeapBeyond · Nov 1, 2020 · f1b29cd · f1b29cd
1 parent 3822cf9
commit f1b29cd
Showing 1 changed file with 4 additions and 4 deletions.
diff --git a/tests/test_detector_named_entity.py b/tests/test_detector_named_entity.py
@@ -3,8 +3,6 @@
 
 from scrubadub.detectors import NamedEntityDetector
 from scrubadub.filth import NameFilth, OrganizationFilth, NamedEntityFilth
-import scrubadub
-
 from base import BaseTestCase
 
 
@@ -15,11 +13,13 @@ class NamedEntityTestCase(unittest.TestCase, BaseTestCase):
     """
 
     def setUp(self):
-        self.detector = NamedEntityDetector()
+        unsupported_version = (sys.version_info.major, sys.version_info.minor) < (3, 6)
         unittest.TestCase.skipTest(
-            (sys.version_info.major, sys.version_info.minor) < (3, 6),
+            unsupported_version,
             "Named entity detector not supported for python<3.6"
         )
+        if not unsupported_version:
+            self.detector = NamedEntityDetector()
 
     def _assert_filth_type_and_pos(self, doc_list, beg_end_list, filth_class):
         doc_names = [str(x) for x in range(len(doc_list))]