From ae0422a2f7451bf7378634054821fed96f519046 Mon Sep 17 00:00:00 2001 From: Tom Close Date: Thu, 30 May 2024 10:13:45 +0930 Subject: [PATCH] Update pydra/utils/tests/test_typing.py Co-authored-by: Chris Markiewicz --- pydra/utils/tests/test_typing.py | 35 ++++++++++++-------------------- 1 file changed, 13 insertions(+), 22 deletions(-) diff --git a/pydra/utils/tests/test_typing.py b/pydra/utils/tests/test_typing.py index 45d1ef46fe..057c44091b 100644 --- a/pydra/utils/tests/test_typing.py +++ b/pydra/utils/tests/test_typing.py @@ -1039,28 +1039,19 @@ def test_type_is_instance11a(): assert not TypeParser.is_instance(None, int | str) -def test_multi_input_obj_coerce1(): - assert TypeParser(MultiInputObj[str])("a") == ["a"] - - -def test_multi_input_obj_coerce2(): - assert TypeParser(MultiInputObj[str])(["a"]) == ["a"] - - -def test_multi_input_obj_coerce3(): - assert TypeParser(MultiInputObj[ty.List[str]])(["a"]) == [["a"]] - - -def test_multi_input_obj_coerce3a(): - assert TypeParser(MultiInputObj[ty.Union[int, ty.List[str]]])(["a"]) == [["a"]] - - -def test_multi_input_obj_coerce3b(): - assert TypeParser(MultiInputObj[ty.Union[int, ty.List[str]]])([["a"]]) == [["a"]] - - -def test_multi_input_obj_coerce4(): - assert TypeParser(MultiInputObj[ty.Union[int, ty.List[str]]])([1]) == [1] +@pytest.mark.parametrize( + ("typ", "obj", "result"), + [ + (MultiInputObj[str], "a", ["a"]), + (MultiInputObj[str], ["a"], ["a"]), + (MultiInputObj[ty.List[str]], ["a"], [["a"]]), + (MultiInputObj[ty.Union[int, ty.List[str]]], ["a"], [["a"]]), + (MultiInputObj[ty.Union[int, ty.List[str]]], [["a"]], [["a"]]), + (MultiInputObj[ty.Union[int, ty.List[str]]], [1], [1]), + ] +) +def test_multi_input_obj_coerce(typ, obj, result): + assert TypeParser(typ)(obj) == result def test_multi_input_obj_coerce4a():