bugfix to unit test

spacetelescope · Jun 12, 2024 · ff7ed8e · ff7ed8e
1 parent 25ab568
commit ff7ed8e
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/jwst/badpix_selfcal/tests/test_badpix_selfcal.py b/jwst/badpix_selfcal/tests/test_badpix_selfcal.py
@@ -249,12 +249,12 @@ def test_badpix_selfcal_step(request, dset):
     assert result[0].meta.cal_step.badpix_selfcal == "COMPLETE"
     if dset == "sci":
         assert len(result) == 2
-        assert len(result[0]) == 1
+        assert isinstance(result[0], dm.IFUImageModel)
         assert len(result[1]) == 0
     else:
         # should return sci, (bkg0, bkg1) but not selfcal0, selfcal1
         assert len(result) == 2
-        assert len(result[0]) == 1
+        assert isinstance(result[0], dm.IFUImageModel)
         assert len(result[1]) == 2