def test_num_cards(self):
""" Test number of cards in deck is correct """
d1, d13 = Deck(), Deck()
d1.reload()
d13.reload(13)
d1_len, d13_len = len(d1.cards), len(d13.cards)
self.assertEqual(d1_len, 52)
self.assertEqual(d13_len, 676)
def test_draw(self):
""" Test draw function draws one card and reloads when no cards are left """
deck = Deck()
deck.reload()
draw_pile = []
for _ in range(0, 52):
draw_pile.append(deck.draw())
self.assertEqual(len(draw_pile), 52)
deck.draw()
self.assertEqual(len(deck.cards), 51)
