Added new swarmplot related unittests

Author: sunroofgod

nbs/tests/test_plot_tools.py

@@ -37,7 +37,7 @@ def test_width_determine():
     ("ax", None, "`ax` must be a Matplotlib AxesSubplot. The current `ax` is a <class 'NoneType'>", ValueError),
     ("order", 5, "`order` must be either an Iterable or None.", ValueError),
     ("hue", 5, "`hue` must be either a string or None.", ValueError),
-    ("palette", None, "`palette` must be either a string or an Iterable.", ValueError),
+    ("palette", None, "`palette` must be either a string indicating a color name or an Iterable.", ValueError),
     ("zorder", None, "`zorder` must be a scalar or float.", ValueError),
     ("size", None, "`size` must be a scalar or float.", ValueError),
     ("side", None, "Invalid `side`. Must be one of 'center', 'right', or 'left'.", ValueError),
@@ -51,12 +51,11 @@ def test_width_determine():
     ("hue", "c", "c is not a column in `data`.", IndexError),
     ("order", ["Control 1", "Test 2"], "Test 2 in `order` is not in the 'group' column of `data`.", IndexError),
     ("palette", " ", "`palette` cannot be an empty string. It must be either a string indicating a color name or an Iterable.", ValueError),
-    ("palette", {"Control 1": " "}, "The color mapping for Control 3 in `palette` is an empty string. It must contain a color name.", ValueError),
-    ("palette", {"Control 1": "black"}, "Control 3 in `palette` is not in the 'group' column of `data`.", IndexError),
+    ("palette", {"Control 1": " "}, "The color mapping for Control 1 in `palette` is an empty string. It must contain a color name.", ValueError),
+    ("palette", {"Control 3": "black"}, "Control 3 in `palette` is not in the 'group' column of `data`.", IndexError),
     # TODO: to add palette validation testing for when color_col is hue
     ("side", "top", "Invalid `side`. Must be one of 'center', 'right', or 'left'.", ValueError)
 ])
-# fmt: on
 def test_swarmplot_input_error_handling(param_name, param_value, error_msg, error_type):
     with pytest.raises(error_type) as excinfo:
         my_data = swarmplot(
@@ -78,7 +77,6 @@ def test_swarmplot_input_error_handling(param_name, param_value, error_msg, erro
     assert error_msg in str(excinfo.value)
 
 
-# fmt: on
 def test_swarmplot_warnings():
     warning_msg = (
         "{0:.1f}% of the points cannot be placed. "
@@ -106,37 +104,33 @@ def test_swarmplot_order_params():
     # `order` should be able to handle customised order -> swapping of params in `order` list
     swarmplot(order=["Control 1", "Test 1"], **default_swarmplot_kwargs)
     swarmplot(order=["Test 1", "Control 1"], **default_swarmplot_kwargs)
+
     # `order` should be able to handle None, where it will then be autogenerated
     swarmplot(order=None, **default_swarmplot_kwargs)
 
 
 def test_swarmplot_hue_params():
     swarmplot(hue="gender", **default_swarmplot_kwargs)
 
-
-def test_swarmplot_palette_params():
+@pytest.mark.parametrize("hue, palette", [
     # `palette` can be a string, list, tuple or a dict
     # Testing `palette` when color of swarms is based on `x` value
-    swarmplot(hue=None, palette="black", **default_swarmplot_kwargs)
-    swarmplot(hue=None, palette=["", "red"], **default_swarmplot_kwargs)
-    swarmplot(hue=None, palette=("black", "red"), **default_swarmplot_kwargs)
-    swarmplot(
-        hue=None,
-        palette={"Control 1": "black", "Test 1": "red"},
-        **default_swarmplot_kwargs
-    )
+    (None, "black"),
+    (None, ("black", "red")),
+    (None, {"Control 1": "black", "Test 1": "red"}),
+
     # Testing `palette` when color of swarms is based on `hue` value
-    swarmplot(hue="gender", palette="black", **default_swarmplot_kwargs)
-    swarmplot(hue="gender", palette=["black", "red"], **default_swarmplot_kwargs)
-    swarmplot(hue="gender", palette=("black", "red"), **default_swarmplot_kwargs)
-    swarmplot(
-        hue="gender",
-        palette={"Female": "black", "Male": "red"},
-        **default_swarmplot_kwargs
-    )
+    ("gender", "black"),
+    ("gender", ["black", "red"]),
+    ("gender", ("black", "red")),
+    ("gender", {"Female": "black", "Male": "red"}),
+
     # Testing auto assignment of `palette` when `palette` is:
     # (list | tuple) and len(palette) != len(unique_color_groups)
-    swarmplot(hue=None, palette=["black"], **default_swarmplot_kwargs)
+    (None, ["black"]),
+])
+def test_swarmplot_palette_params(hue, palette):
+    swarmplot(hue=hue, palette=palette, **default_swarmplot_kwargs)
 
 
 def test_swarmplot_side_params():