def setUp(self): self.shoe1 = BJShoe(1) self.shoe2 = BJShoe(2) self.shoe4 = BJShoe(4) self.shoe6 = BJShoe(6) self.shoe8 = BJShoe(8) self.shoe9 = BJShoe(9)
class TestShoe(unittest.TestCase): def setUp(self): self.shoe1 = BJShoe(1) self.shoe2 = BJShoe(2) self.shoe4 = BJShoe(4) self.shoe6 = BJShoe(6) self.shoe8 = BJShoe(8) self.shoe9 = BJShoe(9) def testRemaining(self): self.assertEqual(self.shoe1.remaining(), 52) self.assertEqual(self.shoe2.remaining(), 52*2) self.assertEqual(self.shoe4.remaining(), 52*4) self.assertEqual(self.shoe6.remaining(), 52*6) self.assertEqual(self.shoe8.remaining(), 52*8) self.assertEqual(self.shoe9.remaining(), 52*9) def testOriginalSize(self): self.assertEqual(self.shoe1.remaining(), 52) self.assertEqual(self.shoe1.original_size, 52) self.shoe1.pop() self.shoe1.pop() self.shoe1.pop() self.assertEqual(self.shoe1.remaining(), 49) self.assertEqual(self.shoe1.original_size, 52) def testRemove(self): self.assertEqual(self.shoe1.remaining(), 52) c1 = self.shoe1.remove('A') self.assertEqual(c1.rank, 'A') c1 = self.shoe1.remove('A') self.assertEqual(c1.rank, 'A') c1 = self.shoe1.remove('A') self.assertEqual(c1.rank, 'A') c1 = self.shoe1.remove('A') self.assertEqual(c1.rank, 'A') c1 = self.shoe1.remove('A') self.assertTrue(c1 is None)