adjusting from statment to reference package name in all test files

DSCI-310-2024 · Apr 11, 2024 · 669bebc · 669bebc
1 parent 1c5d281
commit 669bebc
Show file tree

Hide file tree

Showing 3 changed files with 4 additions and 6 deletions.
diff --git a/tests/test_classification_metrics.py b/tests/test_classification_metrics.py
@@ -3,8 +3,8 @@
 import sys
 import os
 
-sys.path.append(os.path.join(os.path.dirname(__file__), '..'))
-from src.function_classification_metrics import calculate_classification_metrics
+
+from src.py_predpurchase import calculate_classification_metrics
 
 def test_perfect_predictions():
     """

diff --git a/tests/test_feature_importance.py b/tests/test_feature_importance.py
@@ -7,8 +7,7 @@
 from sklearn.ensemble import RandomForestClassifier
 from sklearn.linear_model import LinearRegression
 
-sys.path.append(os.path.join(os.path.dirname(__file__), '..'))
-from src.function_feature_importance import get_feature_importances 
+from src.py_predpurchase import get_feature_importances 
 
 # creating a mock model
 

diff --git a/tests/test_model_cross_val.py b/tests/test_model_cross_val.py
@@ -3,8 +3,7 @@
 import sys
 import os
 
-sys.path.append(os.path.join(os.path.dirname(__file__), '..'))
-from src.function_model_cross_val import model_cross_validation
+from src.py_predpurchase import model_cross_validation
 
 @pytest.fixture
 def test_data():