def test_evaluate_player(self):
game = GameState(initial_cash_for_players=400)
game.add_player(player_name='Cake')
self.assertEqual(400, game.evaluate_player('Cake'))
game.par_company(company_name='NYC', price=50)
game.buy_shares(player_name='Cake',
company_name='NYC',
price=50,
quantity=2)
self.assertEqual(400, game.evaluate_player('Cake'))
game.set_company_price(company_name='NYC', new_price=100)
self.assertEqual(500, game.evaluate_player('Cake'))
game.sell_shares(player_name='Cake',
company_name='NYC',
price=200,
quantity=1)
self.assertEqual(600, game.evaluate_player('Cake'))
def test_evaluate_player_not_found(self):
game = GameState()
with self.assertRaises(ValueError):
game.evaluate_player(player_name='Canelinha')