def __init__(self):
self.reply = ''
self.seat = ''
self.colors = ()
self.hand = []
self.flags = Flags()
self.flag_statuses = ['unclaimed'] * Flags.NUM_FLAGS
self.opponents_last_play = {}
def __stage_mock_state(self):
self.mock_state.seat = 'north'
self.mock_state.colors = ['color{}'.format(num) for num in range(1, 7)]
self.mock_state.hand = [
Card('color1', number).value for number in range(1, 8)
]
self.mock_state.flags = Flags()
class State(object):
NAME = 'random_starterbot'
def __init__(self):
self.reply = ''
self.seat = ''
self.colors = ()
self.hand = []
self.flags = Flags()
self.flag_statuses = ['unclaimed'] * Flags.NUM_FLAGS
self.opponents_last_play = {}
def update_flag_cards(self, flag, seat, cards):
self.flags.add_cards(flag, seat, cards)
def get_full_flags(self):
return [flag + 1 for flag in range(Flags.NUM_FLAGS) if len(self.flags.sides[self.seat][flag]) == 3]
class TestState(unittest.TestCase):
def setUp(self):
self.flags = Flags()
self.flag_set = [[] for _ in range(Flags.NUM_FLAGS)]
self.card_list = [Card('puse', 1).value]
def test_flags_NUM_FLAGS(self):
self.assertEqual(9, Flags.NUM_FLAGS)
def test_init_flags_has_north_flags(self):
self.assertEqual(self.flag_set, self.flags.north)
def test_init_flags_has_south_flags(self):
self.assertEqual(self.flag_set, self.flags.south)
def test_flags_add_a_cards_north(self):
self.flags.add_cards(1, 'north', self.card_list)
self.assertEqual(self.card_list, self.flags.north[0])
def test_flags_add_a_cards_south(self):
self.flags.add_cards(1, 'south', self.card_list)
self.assertEqual(self.card_list, self.flags.south[0])
def __add_three_cards_to_flags(self, flags_list):
self.mock_state.flags = Flags()
for flag in flags_list:
self.mock_state.flags.add_cards(flag, 'north', self.card_list)
def setUp(self):
self.flags = Flags()
self.flag_set = [[] for _ in range(Flags.NUM_FLAGS)]
self.card_list = [Card('puse', 1).value]