From 1264556b59abfcbed134ccbc99d32c02fab97655 Mon Sep 17 00:00:00 2001 From: Boian Petkantchin Date: Fri, 25 Mar 2022 12:43:39 -0700 Subject: [PATCH] Add expected warning in test_wrapped_set_benchmark --- tests/wrappers/core_wrappers_test.py | 6 +++++- 1 file changed, 5 insertions(+), 1 deletion(-) diff --git a/tests/wrappers/core_wrappers_test.py b/tests/wrappers/core_wrappers_test.py index a13dcd8a16..5ddf37e045 100644 --- 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,10 @@ 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, "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"