made changes to test_Prediction

test/test_Prediction.py

@@ -3,7 +3,6 @@ import pandas as pd
 from unittest.mock import patch, Mock
 
 
-# Does the function know 
 def test_numpy_to_pandas():
     from sklearn.datasets import fetch_california_housing
     cal_housing = fetch_california_housing()
@@ -20,9 +19,9 @@ def test_accuracy_metrics2():
     values = tdap.accuracy_metrics({}, None)
     assert values == None
     return 
-    
+
 def test_descriptive_statistic():
-    values = tdap.descriptive_statistic(df, 5)
-    df = pd.DataFrame({'a':[1,2,3,4,5]})
-    assert values == df.shape
+    data = pd.DataFrame({'a':[1,2,3,4,5]})
+    values = tdap.descriptive_statistic(data, 5)
+    assert values == 5
     return 
\ No newline at end of file