- # Make sure nothing else has changed.
- state_1.pop('health')
- state_2.pop('health')
- self.assertEqual(state_1, state_2)
+ self.assert_state(state_1, state_2, exclude=['health'])
+
+ def test_gain_health(self):
+ board = GameBoard.new_game([{'actions': []}])
+ board.health = 2
+ self.assertEqual(board.health, 2)
+ state_1 = board.export()
+
+ board.gain_health()
+ self.assertEqual(board.health, 3)
+ state_2 = board.export()
+
+ self.assert_state(state_1, state_2, exclude=['health'])
+
+ def test_gain_health_at_max(self):
+ board = GameBoard.new_game([{'actions': []}])
+ self.assertEqual(board.health, 4)
+ state_1 = board.export()
+
+ board.gain_health()
+ self.assertEqual(board.health, 4)
+ state_2 = board.export()
+
+ self.assert_state(state_1, state_2)