fix tests for including latest yolo instead of previous models

surfriderfoundationeurope · Mar 2, 2023 · ffece51 · ffece51
1 parent 8e32c5b
commit ffece51
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git a/tests/test_plasticorigins/detection/test_yolo.py b/tests/test_plasticorigins/detection/test_yolo.py
@@ -36,5 +36,5 @@ def test_detect():
     assert len(res) == 3
     assert res[0].shape == (2, 4)
     np.testing.assert_array_equal(
-        res[0], np.array([[200, 233, 56, 233], [302, 331, 26, 99]])
+        res[0], np.array([[202, 233,  56, 244], [301, 325,  28, 106]])
     )
diff --git a/tests/test_plasticorigins/serving/test_inference.py b/tests/test_plasticorigins/serving/test_inference.py
@@ -8,7 +8,7 @@
 
 from plasticorigins.tools.files import create_unique_folder
 from plasticorigins.serving.app import app
-from plasticorigins.serving.config import config_track
+from plasticorigins.serving.config import config_track_yolo as config_track
 from plasticorigins.serving.inference import track
 
 video_file = "tests/ressources/validation_videos/T1_trim.mp4"
@@ -84,7 +84,7 @@ def test_track():
     os.remove(full_filepath)
     assert math.isclose(fps, 5.921, abs_tol=0.01)
     assert num_frames == 34
-    assert len(filtered_results) == 16
+    assert len(filtered_results) == 23
     assert len(filtered_results[0]) == 6
     assert type(filtered_results[0][0]) == int
     assert type(filtered_results[0][1]) == int