Checkpoint 8

tests/test_env.py

@@ -38,6 +38,14 @@ def test_env_reset_determinism_same_seed():
     assert np.allclose(obs_a, obs_b)
 
 
+def test_env_constructor_seed_applies_to_first_reset():
+    a = HerdingEnv(n_sheep=3, seed=42, use_lidar=False)
+    b = HerdingEnv(n_sheep=3, seed=42, use_lidar=False)
+    obs_a, _ = a.reset()
+    obs_b, _ = b.reset()
+    assert np.allclose(obs_a, obs_b)
+
+
 def test_env_curriculum_samples_full_range():
     env = HerdingEnv(seed=0, use_lidar=False)
     sizes = set()