Mock pexpect context management. Fix tests

mlcommons · Jul 28, 2023 · 2a48d2f · 2a48d2f
1 parent 3326768
commit 2a48d2f
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 1 deletion.
diff --git a/cli/medperf/tests/entities/test_cube.py b/cli/medperf/tests/entities/test_cube.py
@@ -133,7 +133,7 @@ def test_cube_runs_command_with_extra_args(self, mocker, setup, task):
         spy = mocker.patch("pexpect.spawn", side_effect=mpexpect.spawn)
         expected_cmd = (
             f"mlcube run --mlcube={self.manifest_path} --task={task} "
-            + f'--platform={self.platform} test="test"'
+            + f'--platform={self.platform} --network=none test="test"'
         )
 
         # Act

diff --git a/cli/medperf/tests/mocks/pexpect.py b/cli/medperf/tests/mocks/pexpect.py
@@ -2,6 +2,12 @@ class MockChild:
     def __init__(self, exitstatus):
         self.exitstatus = exitstatus
 
+    def __enter__(self):
+        pass
+
+    def __exit__(self):
+        self.close()
+
     def isalive(self):
         return False