From 72dc24847e783d7c10b64b7a64cf1aa6995b1a55 Mon Sep 17 00:00:00 2001 From: huangshiyu Date: Tue, 2 Jan 2024 13:18:50 +0800 Subject: [PATCH] update --- tests/test_env/test_offline_env.py | 11 ++++++++--- 1 file changed, 8 insertions(+), 3 deletions(-) diff --git a/tests/test_env/test_offline_env.py b/tests/test_env/test_offline_env.py index e13e255..aa56748 100644 --- a/tests/test_env/test_offline_env.py +++ b/tests/test_env/test_offline_env.py @@ -56,7 +56,7 @@ def gen_data(total_episode, data_save_path): @pytest.fixture(scope="function") -def config(request, tmp_path): +def config(tmp_path): total_episode = 5 data_save_path = tmp_path / "data.pkl" gen_data(total_episode, data_save_path) @@ -66,10 +66,15 @@ def config(request, tmp_path): return cfg +@pytest.fixture(scope="function", params=[True, False]) +def asynchronous(request): + return request.param + + @pytest.mark.unittest -def test_offline_env(config): +def test_offline_env(asynchronous, config): # create environment - env = make("OfflineEnv", env_num=1, cfg=config, asynchronous=True) + env = make("OfflineEnv", env_num=2, cfg=config, asynchronous=asynchronous) for ep_index in range(10): done = False