repair test_UNet2DNucleiBroad_model release

packages/UNet2DNucleiBroad.model/Sigmoid.transformation.yaml

@@ -1,7 +1,6 @@
 name: Sigmoid
 description: Sigmoid activation function
 cite: []
-      doi: TODO
 authors:
     - Constantin Pape
 documentation: transformations.md

tests/test_packages/test_UNet2DNucleiBroad_model.py

@@ -0,0 +1,46 @@
+from pathlib import Path
+from tempfile import TemporaryDirectory
+from typing import List, Optional, Sequence
+from zipfile import ZipFile
+
+import pytest
+
+from pybio.spec.utils import cache_uri, get_instance, load_model
+from tiktorch.server.reader import MODEL_EXTENSIONS
+
+UNET_2D_NUCLEI_BROAD_PACKAGE_URL = (
+    "https://github.com/bioimage-io/pytorch-bioimage-io/releases/download/v0.1/UNet2DNucleiBroad.model.zip"
+)
+
+
+def guess_model_path(file_names: List[str]) -> Optional[str]:
+    for file_name in file_names:
+        if file_name.endswith(MODEL_EXTENSIONS):
+            return file_name
+
+    return None
+
+
+def eval_model_zip(model_zip: ZipFile, cache_path: Path):
+    with TemporaryDirectory() as tempdir:
+        temp_path = Path(tempdir)
+        if cache_path is None:
+            cache_path = temp_path / "cache"
+
+        model_zip.extractall(temp_path)
+        spec_file_str = guess_model_path([str(file_name) for file_name in temp_path.glob("*")])
+        pybio_model = load_model(spec_file_str, root_path=temp_path, cache_path=cache_path)
+
+        return get_instance(pybio_model)
+
+
+@pytest.fixture
+def unet_2d_nuclei_broad_package_bytes(cache_path):
+    return cache_uri(uri_str=UNET_2D_NUCLEI_BROAD_PACKAGE_URL, hash={}, cache_path=cache_path)
+
+
+def test_eval_model_zip(unet_2d_nuclei_broad_package_bytes, cache_path):
+    with ZipFile(unet_2d_nuclei_broad_package_bytes) as zf:
+        model = eval_model_zip(zf, cache_path=cache_path)
+
+    assert model