parametrize_with_cases can get argnames as list

Author: saroad2

pytest_cases/case_parametrizer_new.py

@@ -39,7 +39,7 @@
     pass
 
 
-def parametrize_with_cases(argnames,                # type: str
+def parametrize_with_cases(argnames,                # type: Union[str, List[str]]
                            cases=AUTO,              # type: Union[Callable, Type, ModuleRef]
                            prefix=CASE_PREFIX_FUN,  # type: str
                            glob=None,               # type: str

pytest_cases/fixture_parametrize_plus.py

@@ -800,24 +800,25 @@ def _get_argnames_argvalues(argnames=None, argvalues=None, **args):
         # simplify if needed to comply with pytest.mark.parametrize
         if len(argnames) == 1:
             argvalues = [l[0] if not is_marked_parameter_value(l) else l for l in argvalues]
+        return argnames, argvalues
 
     elif isinstance(argnames, string_types):
         # (2) argnames + argvalues, as usual. However **args can also be passed and should be added
         argnames = get_param_argnames_as_list(argnames)
 
-        if argvalues is None:
-            raise ValueError("No argvalues provided while argnames are provided")
+    if argvalues is None:
+        raise ValueError("No argvalues provided while argnames are provided")
 
-        # transform argvalues to a list (it can be a generator)
-        try:
-            argvalues = list(argvalues)
-        except TypeError:
-            raise InvalidParamsList(argvalues)
-
-        # append **args
-        if len(kw_argnames) > 0:
-            argnames, argvalues = cart_product_pytest((argnames, kw_argnames),
-                                                      (argvalues, kw_argvalues))
+    # transform argvalues to a list (it can be a generator)
+    try:
+        argvalues = list(argvalues)
+    except TypeError:
+        raise InvalidParamsList(argvalues)
+
+    # append **args
+    if len(kw_argnames) > 0:
+        argnames, argvalues = cart_product_pytest((argnames, kw_argnames),
+                                                  (argvalues, kw_argvalues))
 
     return argnames, argvalues