fix test for leduc

Former-commit-id: 955039b
datamllab · Apr 7, 2020 · 416b37c · 416b37c
1 parent 3902ab4
commit 416b37c
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 4 deletions.
diff --git a/tests/envs/test_leducholdem_env.py b/tests/envs/test_leducholdem_env.py
@@ -10,7 +10,7 @@ class TestLeducholdemEnv(unittest.TestCase):
  def test_init_game_and_extract_state(self):
  env = rlcard.make('leduc-holdem')
  state, _ = env.init_game()
- self.assertEqual(state['obs'].size, 34)
+ self.assertEqual(state['obs'].size, 49)
  for action in state['legal_actions']:
  self.assertLess(action, env.action_num)
 

diff --git a/tests/games/test_leducholdem_game.py b/tests/games/test_leducholdem_game.py
@@ -20,7 +20,7 @@ def test_init_game(self):
  self.assertEqual(test_id, player_id)
  self.assertIn('raise', state['legal_actions'])
  self.assertIn('fold', state['legal_actions'])
- self.assertIn('check', state['legal_actions'])
+ self.assertIn('call', state['legal_actions'])
 
  def test_step(self):
  game = Game()
@@ -45,7 +45,7 @@ def test_step(self):
 
  # test check
  game.init_game()
- game.step('check')
+ game.step('call')
  game.step('check')
  self.assertEqual(game.round_counter, 1)
 
@@ -95,7 +95,7 @@ def test_player_get_player_id(self):
  def test_is_over(self):
  game = Game()
  game.init_game()
- game.step('check')
+ game.step('call')
  game.step('check')
  game.step('check')
  game.step('check')