fix tests

Kaggle · Nov 9, 2024 · 41b5efc · 41b5efc
1 parent 0795b41
commit 41b5efc
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/kaggle_environments/envs/chess/test_chess.py b/kaggle_environments/envs/chess/test_chess.py
@@ -15,15 +15,15 @@ def test_chess_three_fold():
     json = env.toJSON()
     assert json["name"] == "chess"
     assert json["statuses"] == ["DONE", "DONE"]
-    assert json["rewards"] == [0, 0]
+    assert json["rewards"] == [1.0, 1.0]
 
 def test_chess_100_move_rule():
     env = make("chess", debug=True)
     env.run(["board_shuffle", "board_shuffle"])
     json = env.toJSON()
     assert json["name"] == "chess"
     assert json["statuses"] == ["DONE", "DONE"]
-    assert json["rewards"] == [0, 0]
+    assert json["rewards"] == [1.0, 1.0]
 
 def test_sufficient_material():
     game = Game()