<<<<<<< HEAD
from otree.api import Currency as c, currency_range, SubmissionMustFail, expect
=======
from otree.api import Currency as c, currency_range, SubmissionMustFail
>>>>>>> 0982dc5566ffb7cb2176b6338e4f4497a3d00fc3
from . import pages
from ._builtin import Bot
from .models import Constants


class PlayerBot(Bot):
    cases = [
        {'offer': c(0), 'return': c(0), 'p1_payoff': c(10), 'p2_payoff': c(0)},
        {'offer': c(5), 'return': c(10), 'p1_payoff': c(15), 'p2_payoff': c(5)},
<<<<<<< HEAD
        {'offer': c(10), 'return': c(30), 'p1_payoff': c(30), 'p2_payoff': c(0)},
=======
        {'offer': c(10), 'return': c(30), 'p1_payoff': c(30), 'p2_payoff': c(0)}
>>>>>>> 0982dc5566ffb7cb2176b6338e4f4497a3d00fc3
    ]

    def play_round(self):
        case = self.case
        if self.player.id_in_group == 1:
<<<<<<< HEAD
            yield pages.Send, dict(sent_amount=case['offer'])

        else:
            for invalid_return in [-1, case['offer'] * Constants.multiplier + 1]:
                yield SubmissionMustFail(
                    pages.SendBack, dict(sent_back_amount=invalid_return)
                )
            yield pages.SendBack, dict(sent_back_amount=case['return'])
=======
            yield (pages.Send, {"sent_amount": case['offer']})

        else:
            for invalid_return in [-1, case['offer'] * Constants.multiplier + 1]:
                yield SubmissionMustFail(pages.SendBack,
                                         {'sent_back_amount': invalid_return})
            yield (pages.SendBack, {'sent_back_amount': case['return']})
>>>>>>> 0982dc5566ffb7cb2176b6338e4f4497a3d00fc3

        if self.player.id_in_group == 1:
            expected_payoff = case['p1_payoff']
        else:
            expected_payoff = case['p2_payoff']

<<<<<<< HEAD
        expect(self.player.payoff, expected_payoff)
=======
        assert self.player.payoff == expected_payoff
>>>>>>> 0982dc5566ffb7cb2176b6338e4f4497a3d00fc3