Update tests for jwst/roman

tests/test_jwst.py

@@ -0,0 +1,28 @@
+"""
+Integration tests with JWST pipeline
+"""
+
+from inspect import getmembers, isclass
+
+import pytest
+
+from stpipe.protocols import DataModel, ModelContainer
+
+datamodels = pytest.importorskip("stdatamodels.jwst.datamodels")
+
+
+@pytest.mark.parametrize(
+    "model",
+    [
+        model[1]
+        for model in getmembers(datamodels, isclass)
+        if issubclass(model[1], datamodels.JwstDataModel)
+    ],
+)
+def test_datamodel(model):
+    assert isinstance(model(), DataModel)
+
+
+def test_model_container():
+    jwst = pytest.importorskip("jwst.datamodels")
+    assert isinstance(jwst.ModelContainer(), ModelContainer)

tests/test_protocols.py

@@ -8,21 +8,6 @@
 from stpipe.protocols import DataModel, ModelContainer
 
 
-def test_roman_datamodel():
-    roman_datamodels = pytest.importorskip("roman_datamodels.datamodels")
-    import roman_datamodels.maker_utils as rutil
-
-    roman_image_tree = rutil.mk_level2_image()
-    image_model = roman_datamodels.ImageModel(roman_image_tree)
-    assert isinstance(image_model, DataModel)
-
-
-def test_jwst_datamodel():
-    jwst_datamodel = pytest.importorskip("stdatamodels.jwst.datamodels")
-    image_model = jwst_datamodel.ImageModel()
-    assert isinstance(image_model, DataModel)
-
-
 def _base_methods():
     def crds_observatory(self):
         pass

tests/test_roman.py

@@ -0,0 +1,29 @@
+"""
+Integration tests with JWST pipeline
+"""
+
+from inspect import getmembers, isclass
+
+import pytest
+
+from stpipe.protocols import DataModel, ModelContainer
+
+datamodels = pytest.importorskip("roman_datamodels.datamodels")
+
+
+@pytest.mark.parametrize(
+    "model",
+    [
+        model[1]
+        for model in getmembers(datamodels, isclass)
+        if model[1] != datamodels.DataModel
+        and issubclass(model[1], datamodels.DataModel)
+    ],
+)
+def test_datamodel(model):
+    assert isinstance(model(), DataModel)
+
+
+def test_model_container():
+    romancal = pytest.importorskip("romancal.datamodels")
+    assert isinstance(romancal.ModelContainer(), ModelContainer)

tox.ini

@@ -71,8 +71,8 @@ commands =
     pytest \
     warnings: -W error \
     xdist: -n auto \
-    jwst: --pyargs jwst --ignore-glob=timeconversion --ignore-glob=associations --ignore-glob=*/scripts/* \
-    romancal: --pyargs romancal \
+    jwst: tests/test_jwst.py --pyargs jwst --ignore-glob=timeconversion --ignore-glob=associations --ignore-glob=*/scripts/* \
+    romancal: tests/test_roman.py --pyargs romancal \
     cov: --cov=src/stpipe --cov-config=pyproject.toml --cov-report=term-missing --cov-report=xml \
     {posargs}