feat: adds column tests, fix typos in col test

UBC-MDS · Jan 18, 2022 · d827fab · d827fab
1 parent 99b9ca3
commit d827fab
Showing 1 changed file with 9 additions and 0 deletions.
diff --git a/src/magmaviz/boxplot.py b/src/magmaviz/boxplot.py
@@ -42,6 +42,15 @@ def boxplot(df, x, y, facet=False):
             "'df' should be of type 'pandas.core.frame.DataFrame', a pandas dataframe."
         )
 
+    # checking if x is a column name in the dataframe
+    assert x in list(
+        df.columns
+    ), "This column specified for 'x' does not exist in the dataframe."
+    # checking if y is a column name in the dataframe
+    assert y in list(
+        df.columns
+    ), "This column specified for 'y' does not exist in the dataframe."
+
     # creating titles
     x_title = re.sub(r"[_.,-]", " ", x)