fix tests

Author: FynnBe

tests/build_spec/test_add_weights.py

@@ -31,8 +31,9 @@ def _test_add_weights(model, tmp_path, base_weights, added_weights, **kwargs):
         assert weight.source.exists()
 
     test_res = _test_model(out_path, added_weights)
+    assert all([s["status"] == "passed" for s in test_res])
     test_res = _test_model(out_path)
-    assert test_res["error"] is None
+    assert all([s["status"] == "passed" for s in test_res])
 
     # make sure the weights were cleaned from the cwd
     assert not os.path.exists(os.path.split(weight_path)[1])

tests/build_spec/test_build_spec.py

@@ -152,7 +152,7 @@ def _test_build_spec(
 
     # test inference for the model to ensure that the weights were written correctly
     test_res = _test_model(out_path)
-    assert test_res["error"] is None
+    assert all([s["status"] == "passed" for s in test_res])
 
 
 def test_build_spec_pytorch(any_torch_model, tmp_path):
@@ -209,7 +209,7 @@ def test_build_spec_training_data2(unet2d_nuclei_broad_model, tmp_path):
 
 
 def test_build_spec_parent1(unet2d_nuclei_broad_model, tmp_path):
-    parent = {"uri": "https:/my-parent-model.org"}
+    parent = {"uri": "https://doi.org/10.5281/zenodo.5764892"}
     _test_build_spec(unet2d_nuclei_broad_model, tmp_path / "model.zip", "torchscript", parent=parent)
 
 

tests/test_resource_tests/test_test_model.py

@@ -46,15 +46,15 @@ def test_validation_section_warning(unet2d_nuclei_broad_model, tmp_path: pathlib
 
     model = load_resource_description(unet2d_nuclei_broad_model)
 
-    summary = test_resource(model)[1]
+    summary = test_resource(model)[2]
     assert summary["name"] == "Test documentation completeness."
     assert summary["warnings"] == {"documentation": "No '# Validation' (sub)section found."}
     assert summary["status"] == "passed"
 
     doc_with_validation = tmp_path / "doc.md"
     doc_with_validation.write_text("# Validation\nThis is a section about how to validate the model on new data")
     model.documentation = doc_with_validation
-    summary = test_resource(model)[1]
+    summary = test_resource(model)[2]
     assert summary["name"] == "Test documentation completeness."
     assert summary["warnings"] == {}
     assert summary["status"] == "passed"