repo env

Author: chisingh

.vscode/settings.json

@@ -0,0 +1,7 @@
+{
+    "python.testing.pytestArgs": [
+        "froggy"
+    ],
+    "python.testing.unittestEnabled": false,
+    "python.testing.pytestEnabled": true
+}

froggy/tests/test_workspace.py

@@ -1,4 +1,20 @@
-import pytest
+from unittest.mock import patch
+from froggy.envs import RepoEnv
+from froggy.utils import load_config
+from pathlib import PosixPath
 
+@patch('sys.argv', ['run.py', 'scripts/config.yaml', '--agent', 'cot', '--debug', '-v'])
 def test_workspace():
-    assert 2 + 2 == 4
+    config, args = load_config()
+    config = config[args.agent]
+
+    assert args.config_file == 'scripts/config.yaml'
+    assert args.agent == 'cot'
+    assert args.debug == True
+    assert args.verbose == True
+    
+    env = RepoEnv(**config["env_kwargs"])
+
+    assert isinstance(env.path, PosixPath)
+    assert env.path == PosixPath('data/pytorch')
+