Execute files relative to their directory

Fixes #305

lineapy/cli/cli.py

@@ -1,4 +1,5 @@
 import logging
+import os
 import pathlib
 
 import click
@@ -83,6 +84,11 @@ def linea_cli(
                 "Source code", rich.syntax.Syntax(code, "python")
             )
         )
+
+    # Change the working directory to that of the script,
+    # To pick up relative data paths
+    os.chdir(file_name.parent)
+
     tracer = Tracer(db, SessionType.SCRIPT)
     transform(code, file_name, tracer)
 

tests/__snapshots__/test_end_to_end/TestEndToEnd.test_housing.1.py

@@ -1,6 +1,6 @@
 import pandas as pd
 from sklearn.ensemble import RandomForestClassifier
-assets = pd.read_csv("tests/ames_train_cleaned.csv")
+assets = pd.read_csv("ames_train_cleaned.csv")
 def is_new(col):
     return col > 1970
 assets["is_new"] = is_new(assets["Year_Built"])

tests/__snapshots__/test_end_to_end/TestEndToEnd.test_housing.py

@@ -14,7 +14,7 @@
 alt.data_transformers.enable("json")
 alt.renderers.enable("mimetype")
 
-assets = pd.read_csv("tests/ames_train_cleaned.csv")
+assets = pd.read_csv("ames_train_cleaned.csv")
 
 sns.relplot(data=assets, x="Year_Built", y="SalePrice", size="Lot_Area")
 
@@ -164,7 +164,7 @@ def is_new(col):
         lineno=11,
         col_offset=9,
         end_lineno=11,
-        end_col_offset=52,
+        end_col_offset=46,
         source_code=source_1.id,
     ),
     function_id=CallNode(
@@ -202,10 +202,10 @@ def is_new(col):
                 lineno=11,
                 col_offset=21,
                 end_lineno=11,
-                end_col_offset=51,
+                end_col_offset=45,
                 source_code=source_1.id,
             ),
-            value="tests/ames_train_cleaned.csv",
+            value="ames_train_cleaned.csv",
         ).id
     ],
 )

tests/housing.py

@@ -8,7 +8,7 @@
 alt.data_transformers.enable("json")
 alt.renderers.enable("mimetype")
 
-assets = pd.read_csv("tests/ames_train_cleaned.csv")
+assets = pd.read_csv("ames_train_cleaned.csv")
 
 sns.relplot(data=assets, x="Year_Built", y="SalePrice", size="Lot_Area")
 

tests/test_end_to_end.py

@@ -1,4 +1,5 @@
 import datetime
+import os
 from pathlib import Path
 from tempfile import NamedTemporaryFile
 
@@ -435,7 +436,12 @@ def test_assignment_destructuring(self, execute):
         assert res.values["b"] == 2
 
     def test_housing(self, execute, python_snapshot):
-        code = (Path(__file__).parent / "housing.py").read_text()
+        tests_dir = Path(__file__).parent
+
+        # Change directory to tests dir before executing
+        os.chdir(tests_dir)
+
+        code = (tests_dir / "housing.py").read_text()
         res = execute(code)
         assert res.slice("p value") == python_snapshot
 

tests/test_script.py

@@ -37,12 +37,11 @@ def test_export_slice_housing():
 
 
 def test_kaggle_example1():
-    os.chdir("examples")
 
     subprocess.check_call(
         [
             "lineapy",
-            "kaggle_example1.py",
+            "examples/kaggle_example1.py",
             "--slice",
             "mushroom feature importance",
         ]
@@ -51,12 +50,11 @@ def test_kaggle_example1():
 
 @pytest.mark.xfail(reason="lambdas aren't supported")
 def test_kaggle_example2():
-    os.chdir("examples")
 
     subprocess.check_call(
         [
             "lineapy",
-            "kaggle_example2.py",
+            "examples/kaggle_example2.py",
             "--slice",
             "nn for diabetes",
         ]