fix tests

roboflow · Nov 12, 2024 · 75c88c1 · 75c88c1
1 parent f01a0b8
commit 75c88c1
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/tests/workflows/unit_tests/core_steps/analytics/test_line_counter_v2.py b/tests/workflows/unit_tests/core_steps/analytics/test_line_counter_v2.py
@@ -63,8 +63,8 @@ def test_line_counter() -> None:
     )
 
     # then
-    assert frame1_result == {"count_in": 0, "count_out": 0}
-    assert frame2_result == {"count_in": 1, "count_out": 1}
+    assert frame1_result == {"count_in": 0, "count_out": 0, "detections_in": frame1_detections[[False, False, False, False]], "detections_out": frame1_detections[[False, False, False, False]]}
+    assert frame2_result == {"count_in": 1, "count_out": 1, "detections_in": frame2_detections[[True, False, False, False]], "detections_out": frame2_detections[[False, True, False, False]]}
 
 
 def test_line_counter_no_trackers() -> None: