Added tests to ensure branch coverage

Author: micahkwok

eda_utils_py/eda_utils_py.py

@@ -85,20 +85,20 @@ def cor_map(dataframe, num_col, col_scheme = 'purpleorange'):
 
     # Tests whether input data is of pd.DataFrame type
     if not isinstance(dataframe, pd.DataFrame):
-        raise Exception("The input dataframe must be of pd.DataFrame type")
+        raise TypeError("The input dataframe must be of pd.DataFrame type")
 
     # Tests whether input num_col is of type list
     if not isinstance(num_col, list):
-        raise Exception("The input num_col must be of type list")
+        raise TypeError("The input num_col must be of type list")
 
     # Tests whether values of num_col is of type str 
     for x in num_col:
         if not isinstance(x, str):
-            raise Exception("The type of values in num_col must all be str")
+            raise TypeError("The type of values in num_col must all be str")
 
     # Tests whether input col_scheme is of type str    
     if not isinstance(col_scheme, str):
-        raise Exception("col_scheme must be of type str")
+        raise TypeError("col_scheme must be of type str")
 
     # Tests whether col_scheme is one of three possible options    
     if col_scheme not in ('purpleorange', 'blueorange', 'redblue'):

tests/test_eda_utils_py.py

@@ -1,5 +1,6 @@
 from eda_utils_py import __version__
 from eda_utils_py import eda_utils_py
+from pytest import raises
 import pandas as pd
 
 def test_version():
@@ -42,4 +43,41 @@ def test_cor_map():
     # Tests whether axes is using correct calculated var column as reference
     assert plot_dict['layer'][0]['encoding']['x']['field'] == 'var1', "x should be referring to var1"
     assert plot_dict['layer'][0]['encoding']['y']['field'] == 'var2', "y should be referring to var2"
-    
+    
+    
+    # Testing the Exception Errors
+    data2 = data.copy().to_csv()
+
+    num_col_test1 = (1, 2, 3, 4)
+    num_col_test2 = [1, 2, 3, 'SepalLengthCm']
+    num_col_test3 = ['hi', 'hey', 'hi']
+    num_col_test4 = ['SepalLengthCm', 'SepalWidthCm', 'PetalWidthCm', 'Species']
+    col_scheme_test = 3
+    
+    # Tests whether data is not of dataframe raises TypeError
+    with raises(TypeError):
+        eda_utils_py.cor_map(data2, num_col_test)
+
+    # Tests whether num_col is of not type list raises TypeError
+    with raises(TypeError):
+        eda_utils_py.cor_map(data, num_col_test1)
+
+    # Tests whether contents of num_col is notof type str raises TypeError
+    with raises(TypeError):
+        eda_utils_py.cor_map(data, num_col_test2)
+
+    # Tests whether inputting unallowed col_scheme raises Exception
+    with raises(Exception):
+        eda_utils_py.cor_map(data, num_col_test, 'bluegreen')
+
+    # Tests whether inputting unallowed col_scheme raises TypeError
+    with raises(TypeError):
+        eda_utils_py.cor_map(data, num_col_test, col_scheme_test)
+
+    # Tests whether columns do not exist in raises Exception
+    with raises(Exception):
+        eda_utils_py.cor_map(data, num_col_test3)
+
+    # Tests whether if not all columns in num_col is numeric raises Exception
+    with raises(Exception):
+        eda_utils_py.cor_map(data, num_col_test4)