fix test_test_model

Author: FynnBe

tests/test_resource_tests/test_test_model.py

@@ -1,12 +1,10 @@
 import pathlib
 
-from bioimageio.spec.shared import yaml
-
 
 def test_error_for_wrong_shape(stardist_wrong_shape):
     from bioimageio.core.resource_tests import test_model
 
-    summary = test_model(stardist_wrong_shape)[0]
+    summary = test_model(stardist_wrong_shape)[-1]
     expected_error_message = (
         "Shape (1, 512, 512, 33) of test output 0 'output' does not match output shape description: "
         "ImplicitOutputShape(reference_tensor='input', "
@@ -18,7 +16,7 @@ def test_error_for_wrong_shape(stardist_wrong_shape):
 def test_error_for_wrong_shape2(stardist_wrong_shape2):
     from bioimageio.core.resource_tests import test_model
 
-    summary = test_model(stardist_wrong_shape2)[0]
+    summary = test_model(stardist_wrong_shape2)[-1]
     expected_error_message = (
         "Shape (1, 512, 512, 1) of test input 0 'input' does not match input shape description: "
         "ParametrizedInputShape(min=[1, 80, 80, 1], step=[0, 17, 17, 0])."
@@ -29,15 +27,15 @@ def test_error_for_wrong_shape2(stardist_wrong_shape2):
 def test_test_model(any_model):
     from bioimageio.core.resource_tests import test_model
 
-    summary = test_model(any_model)[0]
-    assert summary["error"] is None
+    summary = test_model(any_model)
+    assert all([s["status"] for s in summary])
 
 
 def test_test_resource(any_model):
     from bioimageio.core.resource_tests import test_resource
 
-    summary = test_resource(any_model)[0]
-    assert summary["error"] is None
+    summary = test_resource(any_model)
+    assert all([s["status"] for s in summary])
 
 
 def test_validation_section_warning(unet2d_nuclei_broad_model, tmp_path: pathlib.Path):