diff --git a/tests/envs/test_checkers.py b/tests/envs/test_checkers.py index 788ea46..80f24cc 100644 --- a/tests/envs/test_checkers.py +++ b/tests/envs/test_checkers.py @@ -1,7 +1,7 @@ import gym import pytest import ma_gym -from pytest_cases import pytest_parametrize_plus, fixture_ref +from pytest_cases import parametrize_plus, fixture_ref @pytest.fixture(scope='module') def env(): @@ -83,7 +83,7 @@ def test_reset_after_episode_end(env): assert env._total_episode_reward == ep_reward test_reset(env) -@pytest_parametrize_plus('env', +@parametrize_plus('env', [fixture_ref(env), fixture_ref(env_full)]) def test_observation_space(env): diff --git a/tests/envs/test_combat.py b/tests/envs/test_combat.py index 4726029..c03891b 100644 --- a/tests/envs/test_combat.py +++ b/tests/envs/test_combat.py @@ -1,7 +1,7 @@ import gym import pytest import ma_gym -from pytest_cases import pytest_parametrize_plus, fixture_ref +from pytest_cases import parametrize_plus, fixture_ref @pytest.fixture(scope='module') @@ -38,7 +38,7 @@ def test_reset_after_episode_end(env): test_reset(env) -@pytest_parametrize_plus('env', +@parametrize_plus('env', [fixture_ref(env)]) def test_observation_space(env): obs = env.reset() diff --git a/tests/envs/test_predatorprey5x5.py b/tests/envs/test_predatorprey5x5.py index b8fe4d6..b133ccc 100644 --- a/tests/envs/test_predatorprey5x5.py +++ b/tests/envs/test_predatorprey5x5.py @@ -1,7 +1,7 @@ import gym import pytest import ma_gym -from pytest_cases import pytest_parametrize_plus, fixture_ref +from pytest_cases import parametrize_plus, fixture_ref @pytest.fixture(scope='module') @@ -54,7 +54,7 @@ def test_reset_after_episode_end(env): test_reset(env) -@pytest_parametrize_plus('env', +@parametrize_plus('env', [fixture_ref(env), fixture_ref(env_full)]) def test_observation_space(env): diff --git a/tests/envs/test_predatorprey7x7.py b/tests/envs/test_predatorprey7x7.py index 249a99f..dae812e 100755 --- a/tests/envs/test_predatorprey7x7.py +++ b/tests/envs/test_predatorprey7x7.py @@ -1,7 +1,7 @@ import gym import pytest import ma_gym -from pytest_cases import pytest_parametrize_plus, fixture_ref +from pytest_cases import parametrize_plus, fixture_ref @pytest.fixture(scope='module') @@ -44,7 +44,7 @@ def test_reset_after_episode_end(env): test_reset(env) -@pytest_parametrize_plus('env', +@parametrize_plus('env', [fixture_ref(env), fixture_ref(env_full)]) def test_observation_space(env): diff --git a/tests/envs/test_switch2.py b/tests/envs/test_switch2.py index 4371f95..8b28e53 100644 --- a/tests/envs/test_switch2.py +++ b/tests/envs/test_switch2.py @@ -1,7 +1,7 @@ import gym import pytest import ma_gym -from pytest_cases import pytest_parametrize_plus, fixture_ref +from pytest_cases import parametrize_plus, fixture_ref @pytest.fixture(scope='module') def env(): @@ -75,7 +75,7 @@ def test_reset_after_episode_end(env): test_reset(env) -@pytest_parametrize_plus('env', +@parametrize_plus('env', [fixture_ref(env), fixture_ref(env_full)]) def test_observation_space(env):