add more unit test for SWD

open-mmlab · Mar 2, 2023 · dc0e23f · dc0e23f
1 parent 55d8ae8
commit dc0e23f
Showing 1 changed file with 6 additions and 0 deletions.
diff --git a/tests/test_metrics/test_swd.py b/tests/test_metrics/test_swd.py
@@ -23,3 +23,9 @@ def test_swd(init_kwargs, preds, gts, results):
         np.testing.assert_almost_equal(out / 100, res / 100, decimal=1)
     swd.reset()
     assert swd._results == []
+
+    swd.add(preds[:50], gts[:50])
+    swd.add(preds[50:], gts[50:])
+    swd_results = swd.compute()
+    for out, res in zip(swd_results.values(), results):
+        np.testing.assert_almost_equal(out / 100, res / 100, decimal=1)