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))