Attempt to create temp depot for test env

test/test_env.py

@@ -2,17 +2,24 @@
 
 import unittest
 from pysr import julia_helpers
+from tempfile import TemporaryDirectory
 
 
 class TestJuliaProject(unittest.TestCase):
     def test_custom_shared_env(self):
         """Test that we can use PySR in a custom shared env"""
-        test_env_name = "@pysr_test_env"
-        julia_helpers.install(julia_project=test_env_name, quiet=True)
-        Main = julia_helpers.init_julia(julia_project=test_env_name)
-        Main.eval("using SymbolicRegression")
-        Main.eval("using Pkg")
-        cur_project_dir = Main.eval("splitdir(dirname(Base.active_project()))[1]")
-        potential_shared_project_dirs = Main.eval("Pkg.envdir.(DEPOT_PATH)")
-        self.assertIn(cur_project_dir, potential_shared_project_dirs)
-        # TODO: Delete the env at the end.
+        with TemporaryDirectory() as tmpdir:
+            # Create a temp depot to store julia packages (and our custom env)
+            Main = julia_helpers.init_julia()
+            Main.eval(
+                f'pushfirst!(DEPOT_PATH, "{julia_helpers._escape_filename(tmpdir)}")'
+            )
+            test_env_name = "@pysr_test_env"
+            julia_helpers.install(julia_project=test_env_name, quiet=True)
+            Main = julia_helpers.init_julia(julia_project=test_env_name)
+            Main.eval("using SymbolicRegression")
+            Main.eval("using Pkg")
+            cur_project_dir = Main.eval("splitdir(dirname(Base.active_project()))[1]")
+            potential_shared_project_dirs = Main.eval("Pkg.envdir(DEPOT_PATH[1])")
+            self.assertEqual(cur_project_dir, potential_shared_project_dirs)
+            Main.eval("pop!(DEPOT_PATH)")