Fix path

climateintelligence · Apr 5, 2024 · a6b61f0 · a6b61f0
1 parent bf61c77
commit a6b61f0
Showing 1 changed file with 6 additions and 2 deletions.
diff --git a/tests/test_hawk.py b/tests/test_hawk.py
@@ -2,6 +2,8 @@
 
 """Tests for `hawk` package."""
 
+import os
+
 import pandas as pd
 import pytest
 from click.testing import CliRunner  # noqa: F401
@@ -28,8 +30,8 @@ def test_content(response):
 
 
 def test_causal_analysis():
-    df_train = pd.read_csv("hawk/analysis/demo/Ticino_train.csv", header=0)
-    df_test = pd.read_csv("hawk/analysis/demo/Ticino_test.csv", header=0)
+    df_train = pd.read_csv("hawk/demo/Ticino_train.csv", header=0)
+    df_test = pd.read_csv("hawk/demo/Ticino_test.csv", header=0)
     target_column_name = "target"
     pcmci_test_choice = "ParCorr"
     pcmci_max_lag = 0
@@ -54,3 +56,5 @@ def test_causal_analysis():
     )
 
     causal_analysis.run()
+
+    os.system("rm -r tests/output")