def test_trip_hands(self): state = GameState() card_piles = [[[Card("K", "h")], [Card("K", "d")], [Card("K", "s")]], [[Card("3", "c")], [Card("A", "s")], [Card("A", "d")]], [[Card("3", "s")], [Card("A", "c")], [Card("K", "c")]]] state.start_new_game(lucky_card=Card("7", "h"), card_piles=card_piles) trip_hands = state._get_trip_hands() assert len(trip_hands) == 4 assert set([(0, 0), (0, 1), (2, 2)]) in trip_hands assert set([(0, 0), (0, 2), (2, 2)]) in trip_hands assert set([(0, 1), (0, 2), (2, 2)]) in trip_hands assert set([(1, 1), (1, 2), (2, 1)]) in trip_hands # this isn't a valid hand (the cards are on the same row) assert set([(0, 0), (0, 1), (0, 2)]) not in trip_hands