From 41b5efcd600cc6d2adfa9c94d9c133c5254102d6 Mon Sep 17 00:00:00 2001 From: Bovard Doerschuk-Tiberi Date: Sat, 9 Nov 2024 17:08:23 +0000 Subject: [PATCH] fix tests --- kaggle_environments/envs/chess/test_chess.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/kaggle_environments/envs/chess/test_chess.py b/kaggle_environments/envs/chess/test_chess.py index ae56ed79..9c4b7a70 100644 --- a/kaggle_environments/envs/chess/test_chess.py +++ b/kaggle_environments/envs/chess/test_chess.py @@ -15,7 +15,7 @@ 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) @@ -23,7 +23,7 @@ def test_chess_100_move_rule(): 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()