fix test for sys_path in run_in_venv

flojoy-ai · Aug 2, 2023 · 646a7b9 · 646a7b9
1 parent e9baee0
commit 646a7b9
Showing 1 changed file with 6 additions and 3 deletions.
diff --git a/tests/flojoy_node_venv_test_.py b/tests/flojoy_node_venv_test_.py
@@ -49,7 +49,8 @@ def empty_function_with_jax():
     # Test for executable
     assert sys_executable.startswith(mock_venv_cache_dir)
     # Test for sys.path
-    assert sys_path[-1].startswith(mock_venv_cache_dir)
+    assert sys_path[-1].startswith(os.path.dirname(__file__))
+    assert sys_path[-2].startswith(mock_venv_cache_dir)
     # Test for package version
     assert packages_dict["jax"] == "0.4.13"
 
@@ -76,7 +77,8 @@ def empty_function_with_flytekit():
     # Test for executable
     assert sys_executable.startswith(mock_venv_cache_dir)
     # Test for sys.path
-    assert sys_path[-1].startswith(mock_venv_cache_dir)
+    assert sys_path[-1].startswith(os.path.dirname(__file__))
+    assert sys_path[-2].startswith(mock_venv_cache_dir)
     # Test for package version
     assert packages_dict["flytekit"] == "1.8.2"
 
@@ -104,7 +106,8 @@ def empty_function_with_opencv():
     # Test for executable
     assert sys_executable.startswith(mock_venv_cache_dir)
     # Test for sys.path
-    assert sys_path[-1].startswith(mock_venv_cache_dir)
+    assert sys_path[-1].startswith(os.path.dirname(__file__))
+    assert sys_path[-2].startswith(mock_venv_cache_dir)
     # Test for package version
     assert packages_dict["opencv-python-headless"] == "4.7.0.72"