Add expected warning in test_wrapped_set_benchmark

facebookresearch · Mar 26, 2022 · 64f3ef2 · 64f3ef2
1 parent 4715691
commit 64f3ef2
Showing 1 changed file with 6 additions and 1 deletion.
diff --git a/tests/wrappers/core_wrappers_test.py b/tests/wrappers/core_wrappers_test.py
@@ -4,6 +4,7 @@
 # LICENSE file in the root directory of this source tree.
 """Unit tests for //compiler_gym/wrappers."""
 import pytest
+from pytest import warns
 
 from compiler_gym.datasets import Datasets
 from compiler_gym.envs.llvm import LlvmEnv
@@ -182,7 +183,11 @@ def convert_observation(self, observation):
     assert env.benchmark == "benchmark://cbench-v1/dijkstra"
 
     # Repeat again for a different benchmark.
-    env.benchmark = "benchmark://cbench-v1/crc32"
+    with warns(
+        UserWarning,
+        match=r"Changing the benchmark has no effect until reset\(\) is called",
+    ):
+        env.benchmark = "benchmark://cbench-v1/crc32"
     env.reset()
     assert env.benchmark == "benchmark://cbench-v1/crc32"