from . import *
from otree.api import Bot


class PlayerBot(Bot):
    def play_round(self):
        if self.round_number == 1:
            yield (Introduction)
        for i in range(4):
            yield (Main, {'choice': 'Heads'})
            assert (
                self.player.payoff == 1
                if self.player.role == Constants.row_role
                else self.player.payoff == 0
            )
            yield (Results)
        assert (
            self.player.participant.vars['total'] == 4
            if self.player.role == Constants.row_role
            else self.player.participant.vars['total'] == 0
        )
        yield (Final)
        # TODO: assert fails :(