update test for strongsort

mikel-brostrom · Aug 7, 2023 · 12019c1 · 12019c1
1 parent c6bbf64
commit 12019c1
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/tests/test_python.py b/tests/test_python.py
@@ -1,3 +1,5 @@
+# Mikel Broström 🔥 Yolo Tracking 🧾 AGPL-3.0 license
+
 # pytest tests/test_python.py
 
 from pathlib import Path
@@ -192,15 +194,13 @@ def test_strongsort_output():
         half=False,
         per_class=False
     )
-    tracker.n_init = 2
+    tracker.n_init = 1
     rgb = np.random.randint(255, size=(640, 640, 3), dtype=np.uint8)
     det = np.array([[144, 212, 578, 480, 0.82, 0],
                     [425, 281, 576, 472, 0.56, 65]])
     output = tracker.update(det, rgb)
     assert output.size == 0
     output = tracker.update(det, rgb)
-    assert output.size == 0
-    output = tracker.update(det, rgb)
     assert output.shape == (2, 7)  # two inputs should give two outputs
     output = tracker.update(det, rgb)
     assert output.shape == (2, 7)  # two inputs should give two outputs