added samples as a module to conftest

tests/conftest.py

@@ -0,0 +1,19 @@
+import os
+import sys
+from pathlib import Path
+
+
+def add_samples_to_path():
+    samples_dir = str(Path(__file__).parent / "samples")
+
+    if samples_dir not in sys.path:
+        sys.path.append(samples_dir)
+
+    if "PYTHONPATH" in os.environ:
+        os.environ["PYTHONPATH"] = f"{os.environ['PYTHONPATH']}:{samples_dir}"
+    else:
+        os.environ["PYTHONPATH"] = samples_dir
+
+
+def pytest_sessionstart(session):
+    add_samples_to_path()

tests/functions/test_function_deployments.py

@@ -43,7 +43,7 @@
 
 def test_register_no_op_behavior():
     uc = FunctionDeployment("foo", "bar", root_dir=samples_dir)
-    from tests.samples.redact import redact
+    from samples.redact import redact
 
     reg = uc.register(redact)
 
@@ -94,7 +94,7 @@ def test_register_no_op_behavior():
 
 def test_register_secret_behavior():
     uc = FunctionDeployment("foo", "bar", root_dir=samples_dir)
-    from tests.samples.redact_with_secret import redact_w_secret
+    from samples.redact_with_secret import redact_w_secret
 
     reg = uc.register(redact_w_secret)
 
@@ -118,7 +118,7 @@ def test_remote():
         "bar",
         root_dir=samples_dir,
     )
-    from tests.samples.redact import redact
+    from samples.redact import redact
 
     reg = uc.register(redact)
     data = '{"foo": "bar"}'
@@ -137,7 +137,7 @@ def test_remote():
 
 def test_deploy():
     uc = FunctionDeployment("foo", "bar", root_dir=samples_dir)
-    from tests.samples.redact import redact
+    from samples.redact import redact
 
     uc.register(redact)
 

tests/inline/test_recursive_resolver.py

@@ -33,7 +33,7 @@
 
 
 def test_inline():
-    from tests.samples.redact import redact
+    from samples.redact import redact
 
     inline_function(redact, samples_dir)
     assert redact._inlined == True, "Function was not inlined"
@@ -43,7 +43,7 @@ def test_inline():
 
 
 def test_undefined_names():
-    from tests.samples.redact_with_undefined_name import redact_undefined
+    from samples.redact_with_undefined_name import redact_undefined
 
     with pytest.raises(ValueError) as e:
         inline_function(redact_undefined, samples_dir)