class OrderItemTest(unittest.TestCase): def setUp(self): self.item = MenuItem(15, 'Cheeseburger', 'Large', 3.79) def test_GetItemName(self): self.assertEqual(self.item.getItemName(), 'Cheeseburger') def test_GetItemID(self): self.assertEqual(self.item.getItemID(), 15) def test_GetItemSize(self): self.assertEqual(self.item.getItemSize(), 'Large') def test_GetItemPrice(self): self.assertEqual(self.item.getItemPrice(), 3.79)
class MenuItemTest (unittest.TestCase): def setUp(self): self.menuitem = MenuItem('Small',"Burger",3.00,2) def test_GetItemSize(self): self.assertEqual(self.menuitem.getItemSize(), 'small') def test_GetItemName(self): self.assertEqual(self.menuitem.getItemName(), 'Burger') def test_GetItemPrice(self): self.assertEqual(self.menuitem.getItemPrice(), 3.00) def test_GetItemQuantity(self): self.assertEqual(self.menuitem.getItemQuantity(), 2)