fix: add black death wrapper to test.

Farama-Foundation · Nov 7, 2023 · 344ee29 · 344ee29
1 parent 9788c66
commit 344ee29
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/test/test_vector/test_pettingzoo_to_vec.py b/test/test_vector/test_pettingzoo_to_vec.py
@@ -98,6 +98,7 @@ def test_terminal_obs_are_returned():
     """
     max_cycles = 300
     env = knights_archers_zombies_v10.parallel_env(spawn_rate=50, max_cycles=300)
+    env = black_death_v3(env)
     env = pettingzoo_env_to_vec_env_v1(env)
     env.reset(seed=42)