# SimpleDeck Example
# Test the SimpleDeck class using
# traditional tests

from simpledeck import SimpleDeck
import unittest

class MyUnitTestClass(unittest.TestCase):
        
    def test_1(self):
        d = SimpleDeck( )
        self.assertEqual(d.count( ), 20)
        c = d.deal( )
        self.assertEqual(d.count( ), 19)
        self.assertEqual(c, 20)
        d.add_to_bottom(c)
        self.assertEqual(d.count( ), 20)
        d.add_to_bottom(99)
        self.assertEqual(d.count( ), 21)
        self.assertEqual(d.is_empty( ), False)
        while not d.is_empty( ):
            d.deal( )
        self.assertEqual(d.is_empty( ), True)
                  
if __name__ == '__main__':
    unittest.main( )