proroklab
diff --git a/‎tests/test_scenarios/test_football.py
Lines changed: 8 additions & 5 deletions b/‎tests/test_scenarios/test_football.py
Lines changed: 8 additions & 5 deletions
@@ -24,15 +24,18 @@ def setup_env(self, n_envs, **kwargs) -> None:
         )
         self.env.seed(0)
 
-    @pytest.mark.skipif(not sys.platform.startswith("linux"), reason="Test needs linux")
-    def test_ai_vs_random(self, n_envs=4, n_agents=3):
+    @pytest.mark.skipif(
+        sys.platform.startswith("win32"), reason="Test does not work on windows"
+    )
+    def test_ai_vs_random(self, n_envs=4, n_agents=3, scoring_reward=1):
         self.setup_env(
             n_red_agents=n_agents,
             n_blue_agents=n_agents,
             ai_red_agents=True,
             ai_blue_agents=False,
-            dense_reward_ratio=0,
+            dense_reward=False,
             n_envs=n_envs,
+            scoring_reward=scoring_reward,
         )
         all_done = torch.full((n_envs,), False)
         obs = self.env.reset()
@@ -49,10 +52,10 @@ def test_ai_vs_random(self, n_envs=4, n_agents=3):
                     total_rew[:, i] += rews[i]
                 if dones.any():
                     # Done envs should have exactly sum of rewards equal to num_agents
-                    actual_rew = -1 * n_agents
+                    actual_rew = -scoring_reward * n_agents
                     assert torch.equal(
                         total_rew[dones].sum(-1).to(torch.long),
-                        torch.full((dones.sum(),), actual_rew),
+                        torch.full((dones.sum(),), actual_rew, dtype=torch.long),
                     )
                     total_rew[dones] = 0
                     all_done += dones