added tests for single list item splitter

Author: Jeffrey Wigger

pydra/engine/tests/test_helpers_state.py

@@ -31,8 +31,12 @@ def __init__(
     "splitter, keys_exp, groups_exp, grstack_exp",
     [
         ("a", ["a"], {"a": 0}, [[0]]),
+        (["a"], ["a"], {"a": 0}, [[0]]),
+        (("a",), ["a"], {"a": 0}, [[0]]),
         (("a", "b"), ["a", "b"], {"a": 0, "b": 0}, [[0]]),
         (["a", "b"], ["a", "b"], {"a": 0, "b": 1}, [[0, 1]]),
+        ([["a", "b"]], ["a", "b"], {"a": 0, "b": 1}, [[0, 1]]),
+        ((["a", "b"],), ["a", "b"], {"a": 0, "b": 1}, [[0, 1]]),
         ((["a", "b"], "c"), ["a", "b", "c"], {"a": 0, "b": 1, "c": [0, 1]}, [[0, 1]]),
         ([("a", "b"), "c"], ["a", "b", "c"], {"a": 0, "b": 0, "c": 1}, [[0, 1]]),
         ([["a", "b"], "c"], ["a", "b", "c"], {"a": 0, "b": 1, "c": 2}, [[0, 1, 2]]),
@@ -58,6 +62,8 @@ def test_splits_groups(splitter, keys_exp, groups_exp, grstack_exp):
     "keys_final_exp, groups_final_exp, grstack_final_exp",
     [
         ("a", ["a"], ["a"], [], {}, []),
+        (["a"], ["a"], ["a"], [], {}, []),
+        (("a",), ["a"], ["a"], [], {}, []),
         (("a", "b"), ["a"], ["a", "b"], [], {}, [[]]),
         (("a", "b"), ["b"], ["a", "b"], [], {}, [[]]),
         (["a", "b"], ["b"], ["b"], ["a"], {"a": 0}, [[0]]),
@@ -69,6 +75,8 @@ def test_splits_groups(splitter, keys_exp, groups_exp, grstack_exp):
         ([("a", "b"), "c"], ["a"], ["a", "b"], ["c"], {"c": 0}, [[0]]),
         ([("a", "b"), "c"], ["b"], ["a", "b"], ["c"], {"c": 0}, [[0]]),
         ([("a", "b"), "c"], ["c"], ["c"], ["a", "b"], {"a": 0, "b": 0}, [[0]]),
+        ([[("a", "b"), "c"]], ["c"], ["c"], ["a", "b"], {"a": 0, "b": 0}, [[0]]),
+        (([("a", "b"), "c"],), ["c"], ["c"], ["a", "b"], {"a": 0, "b": 0}, [[0]]),
     ],
 )
 def test_splits_groups_comb(
@@ -94,6 +102,8 @@ def test_splits_groups_comb(
     "splitter, cont_dim, values, keys, splits",
     [
         ("a", None, [(0,), (1,)], ["a"], [{"a": 1}, {"a": 2}]),
+        (["a"], None, [(0,), (1,)], ["a"], [{"a": 1}, {"a": 2}]),
+        (("a",), None, [(0,), (1,)], ["a"], [{"a": 1}, {"a": 2}]),
         (
             ("a", "v"),
             None,
@@ -468,6 +478,8 @@ def test_splits_2(splitter_rpn, inner_inputs, values, keys, splits):
         (["a", ("b", ["c", "d"])], ["a", "b", "c", "d", "*", ".", "*"]),
         ((["a", "b"], "c"), ["a", "b", "*", "c", "."]),
         ((["a", "b"], ["c", "d"]), ["a", "b", "*", "c", "d", "*", "."]),
+        (([["a", "b"]], ["c", "d"]), ["a", "b", "*", "c", "d", "*", "."]),
+        (((["a", "b"],), ["c", "d"]), ["a", "b", "*", "c", "d", "*", "."]),
         ([("a", "b"), ("c", "d")], ["a", "b", ".", "c", "d", ".", "*"]),
     ],
 )