Register doudizhu model and fix perfect information game

Former-commit-id: 776a169

rlcard/envs/doudizhu.py

@@ -128,6 +128,6 @@ def get_perfect_information(self):
         state['hand_cards'] = [cards2str(player.current_hand) for player in self.game.players]
         state['landlord'] = self.game.state['landlord']
         state['trace'] = self.game.state['trace']
-        state['curent_player'] = self.game.round.current_player
+        state['current_player'] = self.game.round.current_player
         state['legal_actions'] = self.game.state['actions']
         return state

rlcard/envs/simpledoudizhu.py

@@ -124,6 +124,6 @@ def get_perfect_information(self):
         state['hand_cards'] = [cards2str(player.current_hand) for player in self.game.players]
         state['landlord'] = self.game.state['landlord']
         state['trace'] = self.game.state['trace']
-        state['curent_player'] = self.game.round.current_player
+        state['current_player'] = self.game.round.current_player
         state['legal_actions'] = self.game.state['actions']
         return state

rlcard/models/__init__.py

@@ -37,6 +37,10 @@
     model_id = 'limit-holdem-rule-v1',
     entry_point='rlcard.models.limitholdem_rule_models:LimitholdemRuleModelV1')
 
+register(
+    model_id = 'doudizhu-rule-v1',
+    entry_point='rlcard.models.doudizhu_rule_models:DouDizhuRuleModelV1')
+
 register(
     model_id='gin-rummy-novice-rule',
     entry_point='rlcard.models.gin_rummy_rule_models:GinRummyNoviceRuleModel')

tests/envs/test_doudizhu_env.py

@@ -70,5 +70,9 @@ def test_decode_action(self):
         decoded = env._decode_action(29)
         self.assertEqual(decoded, '444')
 
+    def test_get_perfect_information(self):
+        env = rlcard.make('doudizhu')
+        _, player_id = env.reset()
+        self.assertEqual(player_id, env.get_perfect_information()['current_player'])
 if __name__ == '__main__':
     unittest.main()

tests/envs/test_simpledoudizhu_env.py

@@ -70,5 +70,9 @@ def test_decode_action(self):
         decoded = env._decode_action(14)
         self.assertEqual(decoded, '888')
 
+    def test_get_perfect_information(self):
+        env = rlcard.make('simple-doudizhu')
+        _, player_id = env.reset()
+        self.assertEqual(player_id, env.get_perfect_information()['current_player'])
 if __name__ == '__main__':
     unittest.main()

tests/envs/test_uno_env.py

@@ -77,5 +77,9 @@ def test_single_agent_mode(self):
         for _ in range(100):
             state, _, _ = env.step(np.random.choice(state['legal_actions']))
 
+    def test_get_perfect_information(self):
+        env = rlcard.make('uno')
+        _, player_id = env.reset()
+        self.assertEqual(player_id, env.get_perfect_information()['current_player'])
 if __name__ == '__main__':
     unittest.main()