fix silly problem with unit test

spacetelescope · Oct 8, 2024 · 843c18c · 843c18c
1 parent 622b58a
commit 843c18c
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/tests/test_step.py b/tests/test_step.py
@@ -427,7 +427,11 @@ class StepWithModel(Step):
  def process(self, input_model):
  # make a change to ensure step skip is working
  # without having to define SimpleDataModel.meta.stepname
- input_model.stepstatus = "COMPLETED"
+ if isinstance(input_model, SimpleDataModel):
+ input_model.stepstatus = "COMPLETED"
+ elif isinstance(input_model, SimpleContainer):
+ for model in input_model:
+ model.stepstatus = "COMPLETED"
  return input_model