def test_draw(self):
testDeck = Deck()
card = testDeck.draw()
self.assertEqual(card, ['TEHRAN', 'black'], 'did not draw top card')
self.assertEqual(testDeck._discard_len(), 1,
'discard pile incorrect size')
testDeck.shuffle()
testDeck.draw()
testDeck.draw()
testDeck.draw()
self.assertEqual(testDeck._discard_len(), 4,
'discard pile incorrect size')
def test_shuffle(self):
testDeck = Deck()
testDeck.shuffle()
testDeck.shuffle()
testDeck.shuffle()
self.assertNotEqual(testDeck, None, "shuffle deleted deck")
self.assertEqual(len(testDeck), 48, 'shuffle changed deck size')
self.assertNotEqual(testDeck[0], ['BANGKOK', 'red'],
'card was in same slot after shuffle')
self.assertNotEqual(testDeck[12], ['BOGOTA', 'yellow'],
'card was in same slot after shuffle')
self.assertNotEqual(testDeck[24], ['ATLANTA', 'blue'],
'card was in same slot after shuffle')
self.assertNotEqual(testDeck[47], ['TEHRAN', 'black'],
'card was in same slot after shuffle')