kill process after use (#38)

Lightning-AI · Apr 12, 2024 · ed633aa · ed633aa
1 parent 53588f9
commit ed633aa
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/tests/test_lit_server.py b/tests/test_lit_server.py
@@ -106,7 +106,7 @@ def test_single_loop(simple_litapi, loop_args):
 
 
 def test_run():
-    subprocess.Popen(
+    process = subprocess.Popen(
         ["python", "tests/simple_server.py"],
         stdout=subprocess.DEVNULL,
         stderr=subprocess.DEVNULL,
@@ -118,3 +118,4 @@ def test_run():
     output = subprocess.run("python client.py", shell=True, capture_output=True, text=True).stdout
     assert '{"output":16.0}' in output
     os.remove("client.py")
+    process.kill()