From 536c849c90a2a24ead39b1beff75ab20565829d2 Mon Sep 17 00:00:00 2001 From: Pan Teparak Date: Tue, 22 Oct 2024 13:02:49 +0700 Subject: [PATCH] feat: fix incorrect imports --- tests/test_db_heartbeat.py | 6 +----- 1 file changed, 1 insertion(+), 5 deletions(-) diff --git a/tests/test_db_heartbeat.py b/tests/test_db_heartbeat.py index c49b8fe2..6cc39df1 100644 --- 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()