def test_text2_init(self):
"Test the Vent object creation from text"
# 1. Create Vent object from text
myobj = vent.Vent(text=EXAMPLE_TEXT2)
# 2. Make sure it has the expected values
self.assertEqual(myobj.part2, False)
self.assertEqual(len(myobj.text), 10)
self.assertEqual(len(myobj.locs), 5)
def test_text_init_two_three(self):
"Test the Vent object creation from for part two"
# 1. Create Vent object from text
myobj = vent.Vent(text=EXAMPLE_TWO_THREE, part2=True)
# 2. Make sure it has the expected values
self.assertEqual(myobj.part2, True)
self.assertEqual(len(myobj.text), 10)
self.assertEqual(len(myobj.locs), 9)
def test_empty_init(self):
"Test the default Vent creation"
# 1. Create default Vent object
myobj = vent.Vent()
# 2. Make sure it has the default values
self.assertEqual(myobj.part2, False)
self.assertEqual(myobj.text, None)
self.assertEqual(len(myobj.locs), 0)
def __init__(self, text=None, part2=False):
# 1. Set the initial values
self.part2 = part2
self.text = text
self.vents = []
# 2. Process text (if any)
if text is not None and len(text) > 0:
for line in text:
self.vents.append(vent.Vent(text=line, part2=part2))