cleaning up broken unittests

Author: tclose

pydra/design/tests/test_shell.py

@@ -719,7 +719,9 @@ class Outputs(ShellOutputs):
         "x",
         "y",
     ]
-    assert sorted(a.name for a in attrs.fields(A.Outputs)) == [
+    assert sorted(
+        a.name for a in attrs.fields(A.Outputs) if not a.name.startswith("_")
+    ) == [
         "return_code",
         "stderr",
         "stdout",

pydra/engine/state.py

@@ -866,6 +866,7 @@ def combiner_validation(self):
     def prepare_states(
         self,
         inputs: dict[str, ty.Any],
+        cont_dim: dict[str, int] | None = None,
     ):
         """
         Prepare a full list of state indices and state values.
@@ -881,6 +882,8 @@ def prepare_states(
         self.combiner_validation()
         self.set_input_groups()
         self.inputs = inputs
+        if cont_dim is not None:
+            self.cont_dim = cont_dim
         if self.other_states:
             st: State
             for nm, (st, _) in self.other_states.items():

pydra/engine/tests/test_workflow.py

@@ -2855,8 +2855,7 @@ def Worky2(x, y):
     # checking if the second worky didn't run again
     # checking all directories
 
-    for odir in worky2.output_dir:
-        assert not odir.exists()
+    assert results1.output_dir == results2.output_dir
 
 
 @pytest.mark.flaky(reruns=3)