Fix tests (#304)

* fix tests * remove unused lines of js
Kaggle · Nov 9, 2024 · 271797f · 271797f
1 parent 0795b41
commit 271797f
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 4 deletions.
diff --git a/kaggle_environments/envs/chess/chess.js b/kaggle_environments/envs/chess/chess.js
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()