<<<<<<< HEAD
from otree.api import Currency as c, currency_range, SubmissionMustFail, Submission, expect
=======
from otree.api import (
    Currency as c, currency_range, SubmissionMustFail, Submission
)
>>>>>>> 0982dc5566ffb7cb2176b6338e4f4497a3d00fc3
from . import pages
from ._builtin import Bot
from .models import Constants


class PlayerBot(Bot):

    cases = ['basic', 'min', 'max']

    def play_round(self):
        case = self.case
<<<<<<< HEAD
        yield pages.Introduction
=======
        yield (pages.Introduction)
>>>>>>> 0982dc5566ffb7cb2176b6338e4f4497a3d00fc3

        if case == 'basic':
            if self.player.id_in_group == 1:
                for invalid_contribution in [-1, 101]:
<<<<<<< HEAD
                    yield SubmissionMustFail(
                        pages.Contribute, dict(contribution=invalid_contribution)
                    )

        contribution = dict(min=0, max=100, basic=50)[case]

        yield pages.Contribute, dict(contribution=contribution)

        yield pages.Results
=======
                    yield SubmissionMustFail(pages.Contribute, {
                        'contribution': invalid_contribution})

        contribution = {
            'min': 0,
            'max': 100,
            'basic': 50,
        }[case]

        yield (pages.Contribute, {"contribution": contribution})

        yield (pages.Results)
>>>>>>> 0982dc5566ffb7cb2176b6338e4f4497a3d00fc3

        if self.player.id_in_group == 1:

            if case == 'min':
                expected_payoff = 100
            elif case == 'max':
                expected_payoff = 200
            else:
                expected_payoff = 150
<<<<<<< HEAD
            expect(self.player.payoff, expected_payoff)
=======
            assert self.player.payoff == expected_payoff
>>>>>>> 0982dc5566ffb7cb2176b6338e4f4497a3d00fc3