fixup! [AIRFLOW-6058] Running tests with pytest (#6472)

apache · Dec 10, 2019 · 036165f · 036165f
1 parent 51cca35
commit 036165f
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/tests/contrib/sensors/test_hdfs_sensor.py b/tests/contrib/sensors/test_hdfs_sensor.py
@@ -28,7 +28,7 @@
 
 class HdfsSensorFolderTests(unittest.TestCase):
     def setUp(self):
-        from tests.core import FakeHDFSHook
+        from tests.test_core import FakeHDFSHook
         self.hook = FakeHDFSHook
         self.log = logging.getLogger()
         self.log.setLevel(logging.DEBUG)
@@ -124,7 +124,7 @@ def test_should_be_non_empty_directory_fail(self):
 
 class HdfsSensorRegexTests(unittest.TestCase):
     def setUp(self):
-        from tests.core import FakeHDFSHook
+        from tests.test_core import FakeHDFSHook
         self.hook = FakeHDFSHook
         self.log = logging.getLogger()
         self.log.setLevel(logging.DEBUG)