Refactor for pytest filepaths

environmental_insights/air_pollution_functions.py

@@ -5,7 +5,7 @@
 import matplotlib.patches as mpatches
 from matplotlib import colors
 import numpy as np
-from . import variables
+import environmental_insights.variables as variables  # Absolute import
 from pathlib import Path
 
 pd.options.mode.chained_assignment = None  # default='warn'

environmental_insights/models.py

@@ -1,7 +1,7 @@
 import pickle
 import os
 import pandas as pd
-from . import variables
+import environmental_insights.variables as variables  # Absolute import
 import numpy as np
 
 

tests/test_air_pollution_functions.py

@@ -1,6 +1,9 @@
 import unittest
 import sys
-sys.path.append("../environmental_insights/")  # Adds higher directory to python modules path.
+import os
+
+# Add the parent directory to the system path
+sys.path.insert(0, os.path.abspath(os.path.join(os.path.dirname(__file__), '../environmental_insights')))
 import air_pollution_functions as ei_air_pollution_functions
 import data as ei_data
 

tests/test_data.py

@@ -1,7 +1,12 @@
 import unittest
 import sys
 
-sys.path.append("../environmental_insights/")  # Adds higher directory to python modules path.
+import sys
+import os
+
+# Add the parent directory to the system path
+sys.path.insert(0, os.path.abspath(os.path.join(os.path.dirname(__file__), '../environmental_insights')))
+
 import air_pollution_functions as ei_air_pollution_functions
 import data as ei_data
 

tests/test_models.py

@@ -1,7 +1,19 @@
 import unittest
 import sys
+import os
+
+# Add the parent directory to the system path
+sys.path.insert(0, os.path.abspath(os.path.join(os.path.dirname(__file__), '../environmental_insights')))
+
+import sys
+import os
+
+# Calculate the absolute path to the module's directory
+module_path = os.path.abspath(os.path.join('..', 'environmental_insights'))
+
+# Add this directory to the system path
+sys.path.insert(0, module_path) 
 
-sys.path.append("../environmental_insights/")  # Adds higher directory to python modules path.
 import air_pollution_functions as ei_air_pollution_functions
 import data as ei_data
 import models as ei_models