diff --git a/tests/test_analysis.py b/tests/test_analysis.py index a74abc5..14745b2 100644 --- a/tests/test_analysis.py +++ b/tests/test_analysis.py @@ -137,7 +137,7 @@ class TestFilterHumans(unittest.TestCase): detections = [ [('Bird_A', 0.9), ('Bird_B', 0.8)], [('Bird_D', 0.9), ('Bird_E', 0.8)], - [('Bird_C', 0.7), ('Bird_C', 0.7), ('Bird_C', 0.7), ('Bird_C', 0.7), ('Bird_C', 0.7), ('Bird_C', 0.7), ('Bird_C', 0.7), ('Bird_C', 0.7), ('Bird_C', 0.7), ('Bird_C', 0.5), ('Bird_C', 0.7), ('Human_Human', 0.95)], + [('Bird_C', 0.7)] * 10 + [('Human_Human', 0.95)], [('Bird_F', 0.6), ('Bird_G', 0.5)] ] @@ -145,7 +145,7 @@ class TestFilterHumans(unittest.TestCase): expected = [ [('Bird_A', 0.9), ('Bird_B', 0.8)], [('Bird_D', 0.9), ('Bird_E', 0.8)], - [('Bird_C', 0.7), ('Bird_C', 0.7), ('Bird_C', 0.7), ('Bird_C', 0.7), ('Bird_C', 0.7), ('Bird_C', 0.7), ('Bird_C', 0.7), ('Bird_C', 0.7), ('Bird_C', 0.7), ('Bird_C', 0.5)], + [('Bird_C', 0.7)] * 10, [('Bird_F', 0.6), ('Bird_G', 0.5)] ] @@ -165,7 +165,7 @@ class TestFilterHumans(unittest.TestCase): detections = [ [('Bird_A', 0.9), ('Bird_B', 0.8)], [('Bird_D', 0.9), ('Bird_E', 0.8)], - [('Bird_C', 0.7), ('Bird_C', 0.7), ('Bird_C', 0.7), ('Bird_C', 0.7), ('Bird_C', 0.7), ('Bird_C', 0.7), ('Bird_C', 0.7), ('Bird_C', 0.7), ('Bird_C', 0.7), ('Bird_C', 0.5), ('Bird_C', 0.7), ('Human_Human', 0.95)], + [('Bird_C', 0.7)] * 10 + [('Human_Human', 0.95)], [('Bird_F', 0.6), ('Bird_G', 0.5)] ] @@ -183,5 +183,6 @@ class TestFilterHumans(unittest.TestCase): # Assertions self.assertEqual(result, expected) + if __name__ == '__main__': unittest.main() diff --git a/tests/test_apprise_notifications.py b/tests/test_apprise_notifications.py index b2c3d89..0269aa8 100644 --- a/tests/test_apprise_notifications.py +++ b/tests/test_apprise_notifications.py @@ -10,6 +10,7 @@ from scripts.utils.notifications import sendAppriseNotifications from tests.helpers import Settings + class TestAppriseNotifications(unittest.TestCase): def setUp(self):