def setUp(self):
super().setUp()
self.pencilSet = CompositeItem("Pencil Set", self.pencil,
self.ruler, self.eraser)
self.box = SimpleItem("Box", 1.00)
self.boxedPencilSet = CompositeItem("Boxed Pencil Set",
self.box, self.pencilSet)
self.boxedPencilSet.add(self.pencil)
class CompositeItemTest(SimpleItemTest):
def setUp(self):
super().setUp()
self.pencilSet = CompositeItem("Pencil Set", self.pencil,
self.ruler, self.eraser)
self.box = SimpleItem("Box", 1.00)
self.boxedPencilSet = CompositeItem("Boxed Pencil Set",
self.box, self.pencilSet)
self.marker = SimpleItem("Marker", 0.40)
self.boxedPencilSet.add(self.marker)
def test_create(self):
self.assertEqual("Pencil Set", self.pencilSet.name)
self.assertEqual(3, len(self.pencilSet.children))
self.assertEqual(self.boxedPencilSet.name, "Boxed Pencil Set")
def test_composite(self):
self.assertTrue(self.pencilSet.composite)
self.assertTrue(self.boxedPencilSet.composite)
def test_price(self):
self.assertAlmostEqual(2.20, self.pencilSet.price)
self.assertAlmostEqual(3.60, self.boxedPencilSet.price)
def test_add_simple_item(self):
num_before = len(self.boxedPencilSet.children)
bag = SimpleItem("Bag", 1.00)
self.boxedPencilSet.add(bag)
self.assertEqual(num_before + 1, len(self.boxedPencilSet.children))
def test_remove_simple_item(self):
num_before = len(self.boxedPencilSet.children)
self.boxedPencilSet.remove(self.marker)
self.assertEqual(num_before - 1, len(self.boxedPencilSet.children))
class CompositeItemTest(SimpleItemTest):
def setUp(self):
super().setUp()
self.pencilSet = CompositeItem("Pencil Set", self.pencil, self.ruler,
self.eraser)
self.box = SimpleItem("Box", 1.00)
self.boxedPencilSet = CompositeItem("Boxed Pencil Set", self.box,
self.pencilSet)
self.boxedPencilSet.add(self.pencil)
def test_create(self):
super().test_create()
self.assertEqual(self.boxedPencilSet.name, "Boxed Pencil Set")
self.assertAlmostEqual(self.boxedPencilSet.price, 3.60)
def test_composite(self):
# TODO Implement me!
# Hint: Try asserting that a composite object is really a
# composite object!
# Do not forget to replace self.fail() statement with
# correct assertion method(s)!
self.fail("Not yet implemented.")
class CompositeItemTest(SimpleItemTest):
def setUp(self):
super().setUp()
self.pencilSet = CompositeItem("Pencil Set", self.pencil,
self.ruler, self.eraser)
self.box = SimpleItem("Box", 1.00)
self.boxedPencilSet = CompositeItem("Boxed Pencil Set",
self.box, self.pencilSet)
self.boxedPencilSet.add(self.pencil)
def test_create(self):
super().test_create()
self.assertEqual(self.boxedPencilSet.name, "Boxed Pencil Set")
self.assertAlmostEqual(self.boxedPencilSet.price, 3.60)
def test_composite(self):
# TODO Implement me!
# Hint: Try asserting that a composite object is really a
# composite object!
# Do not forget to replace self.fail() statement with
# correct assertion method(s)!
self.assertEqual(self.pencilSet.composite, True)
self.assertEqual(self.boxedPencilSet.composite, True)
