Set env with MonkeyPatch (#5134)

gradio-app · Aug 8, 2023 · 56b0227 · 56b0227 · vercel · Aug 8, 2023
1 parent 97d804c
commit 56b0227
Showing 1 changed file with 2 additions and 3 deletions.
diff --git a/test/test_blocks.py b/test/test_blocks.py
@@ -434,15 +434,14 @@ def test_raise_error_if_event_queued_but_queue_not_enabled(self):
 
         demo.close()
 
-    def test_concurrency_count_zero_gpu(self):
-        os.environ["SPACES_ZERO_GPU"] = "true"
+    def test_concurrency_count_zero_gpu(self, monkeypatch):
+        monkeypatch.setenv("SPACES_ZERO_GPU", "true")
         demo = gr.Blocks()
         with pytest.warns():
             demo.queue(concurrency_count=42)
         with pytest.warns():
             demo.queue(42)
         assert demo._queue.max_thread_count == demo.max_threads
-        del os.environ["SPACES_ZERO_GPU"]
 
 
 class TestTempFile: