class TraitListTestCase(unittest.TestCase): def setUp(self): self.traits = [] self.traitlist = TraitList() def __build_trait(self, name, val, note=''): t = Trait() t.name = name t.val = val t.note = note return t def testAddition(self): assert self.traitlist.trait_changes == None t = self.__build_trait('Example', 3, 'Carlsbad') self.traitlist.add_trait(t) assert t in self.traitlist.traits assert len(self.traitlist.traits) == 1 assert self.traitlist.trait_changes != None changes = self.traitlist.get_changes_strings() assert changes.first == "Added Example x3 (Carlsbad)" def testDisplay(self): fully_tested = { self.__build_trait('Example', 3, 'Carlsbad') : { '0' : 'Example', '1' : 'Example x3 (Carlsbad)', '2' : 'Example x3 OOO (Carlsbad)', '3' : 'Example OOO (Carlsbad)', '4' : 'Example (3, Carlsbad)', '5' : 'Example (Carlsbad)', '6' : 'Example (3)', '7' : 'Example (Carlsbad)OExample (Carlsbad)OExample (Carlsbad)', '8' : 'OOO', '9' : '3', '10': 'Carlsbad', 'Default': 'Example x3 (Carlsbad)' }, self.__build_trait('Example', 3, '') : { '0' : 'Example', '1' : 'Example x3', '2' : 'Example x3 OOO', '3' : 'Example OOO', '4' : 'Example (3)', '5' : 'Example', '6' : 'Example (3)', '7' : 'ExampleOExampleOExample', '8' : 'OOO', '9' : '3', '10': '', 'Default': 'Example x3' }, self.__build_trait('Example', 0, '') : { '0' : 'Example', '1' : 'Example', '2' : 'Example', '3' : 'Example', '4' : 'Example', '5' : 'Example', '6' : 'Example', '7' : 'Example', '8' : '', '9' : '', '10': '', 'Default': 'Example' }, self.__build_trait('Example', 0, 'Carlsbad') : { '0' : 'Example', '1' : 'Example (Carlsbad)', '2' : 'Example (Carlsbad)', '3' : 'Example (Carlsbad)', '4' : 'Example (Carlsbad)', '5' : 'Example (Carlsbad)', '6' : 'Example', '7' : 'Example (Carlsbad)', '8' : '', '9' : '', '10': 'Carlsbad', 'Default': 'Example (Carlsbad)' } } for trait, test_list in fully_tested.iteritems(): for display_type, expected_result in test_list.iteritems(): print display_type print trait self.assertEqual(trait.display_str(display_type), expected_result) def testBasic(self): assert False