def test_rank_bet(self): with mock.patch('agents.agent.PredictionMarketAdapter', autospec=True) \ as MockPredictionMarket: mock_prediction_market = MockPredictionMarket.return_value account = '42' agent = Agent(account, logging=False) agent.prediction_history = [[75] * NUM_PREDICTIONS, None, None] agent.aggregate_history = [75] * NUM_PREDICTIONS agent.rank_bet() mock_prediction_market.rank.assert_called_once_with(account)
def test_collect_reward(self): with mock.patch('agents.agent.PredictionMarketAdapter', autospec=True) \ as MockPredictionMarket: mock_prediction_market = MockPredictionMarket.return_value account = '42' agent = Agent(account, logging=False) agent.prediction_history = [[75] * NUM_PREDICTIONS, None, None] agent.aggregate_history = [75] * NUM_PREDICTIONS mock_prediction_market.get_winning_tier.return_value = 'top' agent.collect_reward() mock_prediction_market.transfer_reward.assert_called_once_with( account)