Skip to content

Commit dc0e23f

Browse files
committed
add more unit test for SWD
1 parent 55d8ae8 commit dc0e23f

1 file changed

Lines changed: 6 additions & 0 deletions

File tree

tests/test_metrics/test_swd.py

Lines changed: 6 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -23,3 +23,9 @@ def test_swd(init_kwargs, preds, gts, results):
2323
np.testing.assert_almost_equal(out / 100, res / 100, decimal=1)
2424
swd.reset()
2525
assert swd._results == []
26+
27+
swd.add(preds[:50], gts[:50])
28+
swd.add(preds[50:], gts[50:])
29+
swd_results = swd.compute()
30+
for out, res in zip(swd_results.values(), results):
31+
np.testing.assert_almost_equal(out / 100, res / 100, decimal=1)

0 commit comments

Comments
 (0)