diff --git a/tests/contrib/sensors/test_hdfs_sensor.py b/tests/contrib/sensors/test_hdfs_sensor.py index b03b738686ed8..7de01d4b2ddf9 100644 --- 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)