class InventoryTest(unittest.TestCase): def setUp(self): self.menuitem = MenuItem(1, 'item1', 1.50) self.orderitem = OrderItem(self.item, 1) def test_Quantity(self): self.assertEqual(self.menuitem.getQuantity(), 2) def test_item(self): item = self.orderitem.getItem() self.assertEqual(str(item), 'item1')