feat: fix incorrect imports

revsys · Oct 22, 2024 · 536c849 · 536c849
1 parent b10ab34
commit 536c849
Showing 1 changed file with 1 addition and 5 deletions.
diff --git a/tests/test_db_heartbeat.py b/tests/test_db_heartbeat.py
@@ -1,6 +1,6 @@
 import unittest
 
-from health_check.contrib.db_heartbeat import DatabaseHeartBeatCheck
+from health_check.contrib.db_heartbeat.backends import DatabaseHeartBeatCheck
 from health_check.exceptions import ServiceUnavailable
 
 
@@ -16,7 +16,3 @@ def test_check_status_unexpected_result(self):
         health_check = DatabaseHeartBeatCheck()
         with self.assertRaises(ServiceUnavailable):
             health_check.check_status()
-
-
-if __name__ == "__main__":
-    unittest.main()