diff --git a/test/test_mode_2d_statistics.py b/test/test_mode_2d_statistics.py index 42143693..6315d2fb 100644 --- a/test/test_mode_2d_statistics.py +++ b/test/test_mode_2d_statistics.py @@ -1,3 +1,5 @@ + +import os import pytest import numpy as np import pandas as pd @@ -5,6 +7,9 @@ import metcalcpy.util.mode_2d_ratio_statistics as m2rs import metcalcpy.util.mode_2d_arearat_statistics as m2as + +cwd = os.path.dirname(__file__) + def prepare_data(obj_type = "2d"): """ Prepare some data for testing mode ara rations. @@ -12,7 +17,9 @@ def prepare_data(obj_type = "2d"): robust approach would be to use a real MODE export from METviewer. """ - file_path = "test/data/ee_av_input.data" + os.environ['TEST_DIR'] = cwd + + file_path = f"{cwd}/data/ee_av_input.data" df = pd.read_csv(file_path, sep="\t") df["object_type"] = obj_type