Spaces:
Sleeping
Sleeping
File size: 1,160 Bytes
e1e5d77 2128884 e1e5d77 2128884 919b5fc 2128884 919b5fc 2128884 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 |
"""Contains tests for creating and initializing custom Julia projects"""
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"""
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.name)}")'
)
test_env_name = "@pysr_test_env"
julia_helpers.install(julia_project=test_env_name)
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)")
|