<<<<<<< HEAD
from otree.api import Currency as c, currency_range, expect
=======
from otree.api import Currency as c, currency_range
>>>>>>> 0982dc5566ffb7cb2176b6338e4f4497a3d00fc3
from . import pages
from ._builtin import Bot
from .models import Constants

<<<<<<< HEAD

class PlayerBot(Bot):
    cases = ['both_min', 'both_max', 'p1_lower']
=======
class PlayerBot(Bot):
    cases = [
        'both_min',
        'both_max',
        'p1_lower'
    ]
>>>>>>> 0982dc5566ffb7cb2176b6338e4f4497a3d00fc3

    def play_round(self):
        case = self.case

        # start game
<<<<<<< HEAD
        yield pages.Introduction

        if case == 'both_min':
            yield pages.Claim, dict(claim=Constants.min_amount)
            expect(self.player.payoff, Constants.min_amount)
        elif case == 'both_max':
            yield pages.Claim, dict(claim=Constants.max_amount)
            expect(self.player.payoff, Constants.max_amount)
        else:
            if self.player.id_in_group == 1:
                yield pages.Claim, dict(claim=Constants.min_amount)
                expect(self.player.payoff, Constants.min_amount + 2)
            else:
                yield pages.Claim, dict(claim=Constants.min_amount + 1)
                expect(self.player.payoff, Constants.min_amount - 2)

        yield pages.Results
=======
        yield (pages.Introduction)

        if case == 'both_min':
            yield (pages.Claim, {"claim": Constants.min_amount})
            assert self.player.payoff == Constants.min_amount
        elif case == 'both_max':
            yield (pages.Claim, {"claim": Constants.max_amount})
            assert self.player.payoff == Constants.max_amount
        else:
            if self.player.id_in_group == 1:
                yield (pages.Claim, {"claim": Constants.min_amount})
                assert self.player.payoff == Constants.min_amount + 2
            else:
                yield (pages.Claim, {"claim": Constants.min_amount + 1})
                assert self.player.payoff == Constants.min_amount - 2

        yield (pages.Results)
>>>>>>> 0982dc5566ffb7cb2176b6338e4f4497a3d00fc3