from . import *
from otree.api import Bot


class PlayerBot(Bot):

    def play_round(self):
        if self.round_number == 1:
            yield (Introduction)
        if self.player.id_in_group == 1:
            yield (Main, {'effort': 3})
            assert self.player.payoff == 2
        else:
            yield (Main, {'effort': 1})
            assert self.player.payoff == 4
        yield (Results)