# 从左到右测试

from tradeoff import *


class PlayerBot(Bot):
    def play_round(self):
        # 利用不同的round来做随机，只能测试第一个round
        round_number = self.round_number
        if round_number > 1:
            return

        # 自定义table_list
        table_list = [
            # Part I
            1101, 1102, 1103,
            1201, 1202, 1203,

            # Part II
            2101, 2103, 2107, 2109, 2113, 2115,
            2201, 2203, 2207, 2209, 2213, 2215,

            # Part III
            3101, 3102, 3103, 3107, 3108, 3109, 3113, 3114, 3115,
            3201, 3202, 3203, 3207, 3208, 3209, 3213, 3214, 3215
        ]

        self.participant.vars['table_list'] = table_list

        self.participant.vars['num_rounds'] = len(table_list)

        # ======================
        # part I开始
        # ======================

        yield Intro

        # 读取顺序正常
        assert len(self.participant.vars['table_list']) > 0

        n = round_number - 1
        print(n)

        yield P1Left, dict(sp=85)
        yield P1Left, dict(sp=90)
        yield P1Left, dict(sp=100)

        yield P1Right, dict(sp=90)
        yield P1Right, dict(sp=95)
        yield P1Right, dict(sp=105)

        assert self.participant.vars['upper_1201'] == 95
        assert self.participant.vars['upper_1202'] == 110
        assert self.participant.vars['upper_1203'] == 140

        # print(self.participant.vars)

        # ======================
        # part II 左侧开始
        # ======================

        yield P2Left, dict(sp=35)
        expect(self.participant.vars['lower_2101'], 10)

        yield P2Left, dict(sp=55)
        assert self.participant.vars['lower_2103'] == 50

        yield P2Left, dict(sp=40)
        assert self.participant.vars['lower_2107'] == 20

        yield P2Left, dict(sp=55)
        assert self.participant.vars['lower_2109'] == 50

        yield P2Left, dict(sp=40)
        assert self.participant.vars['lower_2113'] == 20

        yield P2Left, dict(sp=56)
        assert self.participant.vars['lower_2115'] == 52

        # ======================
        # part II 右侧开始
        # ======================

        yield P2Right, dict(sp=86)
        expect(self.participant.vars['lower_2201'], 85)
        expect(self.participant.vars['upper_2201'], 95)

        yield P2Right, dict(sp=86)
        expect(self.participant.vars['lower_2203'], 85)
        expect(self.participant.vars['upper_2203'], 95)

        yield P2Right, dict(sp=100)
        expect(self.participant.vars['lower_2207'], 90)
        expect(self.participant.vars['upper_2207'], 110)

        yield P2Right, dict(sp=100)
        expect(self.participant.vars['lower_2209'], 90)
        expect(self.participant.vars['upper_2209'], 110)

        yield P2Right, dict(sp=120)
        expect(self.participant.vars['lower_2213'], 100)
        expect(self.participant.vars['upper_2213'], 140)

        yield P2Right, dict(sp=120)
        expect(self.participant.vars['lower_2215'], 100)
        expect(self.participant.vars['upper_2215'], 140)

        # ======================
        # part III 左侧开始
        # ======================

        # Block 1
        yield P3Left, dict(sp=80)
        assert self.participant.vars['lower_3101'] == 55
        assert self.participant.vars['upper_3101'] == 105

        yield P3Left, dict(sp=95)
        expect(self.participant.vars['lower_3102'], 85)
        assert self.participant.vars['upper_3102'] == 105

        yield P3Left, dict(sp=100)
        assert self.participant.vars['lower_3103'] == 95
        assert self.participant.vars['upper_3103'] == 105

        # Block 2
        yield P3Left, dict(sp=80)
        assert self.participant.vars['lower_3107'] == 70
        assert self.participant.vars['upper_3107'] == 110

        yield P3Left, dict(sp=100)
        assert self.participant.vars['lower_3108'] == 90
        assert self.participant.vars['upper_3108'] == 110

        yield P3Left, dict(sp=105)
        assert self.participant.vars['lower_3109'] == 100
        assert self.participant.vars['upper_3109'] == 110

        # Block 3

        yield P3Left, dict(sp=110)
        assert self.participant.vars['lower_3113'] == 80
        assert self.participant.vars['upper_3113'] == 120

        yield P3Left, dict(sp=110)
        assert self.participant.vars['lower_3114'] == 100
        assert self.participant.vars['upper_3114'] == 120

        yield P3Left, dict(sp=110)
        assert self.participant.vars['lower_3115'] == 112
        assert self.participant.vars['upper_3115'] == 120

        # ======================
        # part III 右侧开始
        # ======================

        # Block 1
        yield P3Right, dict(sp=90)
        assert self.participant.vars['lower_3201'] == 85
        expect(self.participant.vars['upper_3201'], 95)

        yield P3Right, dict(sp=90)
        expect(self.participant.vars['lower_3202'], 85)
        assert self.participant.vars['upper_3202'] == 95

        yield P3Right, dict(sp=100)
        assert self.participant.vars['lower_3203'] == 85
        assert self.participant.vars['upper_3203'] == 95

        # Block 2
        yield P3Right, dict(sp=100)
        assert self.participant.vars['lower_3207'] == 90
        expect(self.participant.vars['upper_3207'], 110)

        yield P3Right, dict(sp=100)
        expect(self.participant.vars['lower_3208'], 90)
        assert self.participant.vars['upper_3208'] == 110

        yield P3Right, dict(sp=100)
        assert self.participant.vars['lower_3209'] == 90
        assert self.participant.vars['upper_3209'] == 110

        # Block 3
        yield P3Right, dict(sp=110)
        assert self.participant.vars['lower_3213'] == 100
        expect(self.participant.vars['upper_3213'], 140)

        yield P3Right, dict(sp=110)
        expect(self.participant.vars['lower_3214'], 100)
        assert self.participant.vars['upper_3214'] == 140

        yield P3Right, dict(sp=110)
        assert self.participant.vars['lower_3215'] == 100
        assert self.participant.vars['upper_3215'] == 140

        yield Results

#