fixing main test (#57)

* fixing main test

* better

Author: TLSDC

tests/test_main.py

@@ -1,22 +1,26 @@
-from pathlib import Path
 import subprocess
+import sys
+from pathlib import Path
+
 import pytest
 
 
 @pytest.mark.pricy
 def test_main_script_execution():
     # this should trigger agent_4o_mini on miniwob_tiny_test unless this was
     # reconfigured differently.
-    script_path = Path(__file__).parent.parent / "main.py"
+    path = Path(__file__).parent.parent / "main.py"
+
+    sys.path.insert(0, str(path.parent))
 
     # just make sure it's in the right state
-    main = __import__(script_path.stem)
+    main = __import__(path.stem)
     assert main.benchmark == "miniwob_tiny_test"
     assert main.reproducibility_mode == False
     assert main.relaunch == False
     assert main.n_jobs <= 10
 
-    result = subprocess.run(["python", script_path], capture_output=True, text=True, timeout=5 * 60)
+    result = subprocess.run(["python", str(path)], capture_output=True, text=True, timeout=5 * 60)
     assert result.returncode == 0