Skip to content

Commit 9c073ce

Browse files
committed
improve output for tesrt_add_weights
1 parent 53e4d62 commit 9c073ce

File tree

1 file changed

+4
-2
lines changed

1 file changed

+4
-2
lines changed

tests/build_spec/test_add_weights.py

Lines changed: 4 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -31,9 +31,11 @@ def _test_add_weights(model, tmp_path, base_weights, added_weights, **kwargs):
3131
assert weight.source.exists()
3232

3333
test_res = _test_model(out_path, added_weights)
34-
assert all([s["status"] == "passed" for s in test_res])
34+
failed = [s for s in test_res if s["status"] != "passed"]
35+
assert not failed, failed
3536
test_res = _test_model(out_path)
36-
assert all([s["status"] == "passed" for s in test_res])
37+
failed = [s for s in test_res if s["status"] != "passed"]
38+
assert not failed, failed
3739

3840
# make sure the weights were cleaned from the cwd
3941
assert not os.path.exists(os.path.split(weight_path)[1])

0 commit comments

Comments
 (0)