def test_create_cards(self): """test the create_cards function""" ranks = create_ranks(['ship', 'shape', 'shop']) suits = create_suits(['x', 'y', 'z']) cards = create_cards(ranks, suits) for card in cards: self.assertEqual(card.__class__, Card) self.assertEqual(card, card) self.assertEqual(card, Card(Rank(card.rank.index, card.rank.label), Suit(card.suit.label))) self.assertEqual(len(ranks)*len(suits), len(cards)) self.assertNotEqual(cards[0], cards[2])
def test_create_ranks(self): """test the create_ranks function""" labels = ['foo', 'bar', 'quux'] ranks = create_ranks(labels) found_bar = False for rank in ranks: self.assertEqual(rank.__class__, Rank) self.assertEqual(rank, rank) self.assertEqual(rank, Rank(rank.index, rank.label)) self.assertEqual(rank.label, str(rank)) if rank.label == 'bar': self.assertEqual(1, rank.index) found_bar = True self.assert_(found_bar) self.assert_(ranks[0] < ranks[2])