def test_no_rank_if_no_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 = [None, None, None] agent.rank_bet() mock_prediction_market.rank.assert_not_called()
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)