def test_retrieve_values_by_key_method(): class TestLoadClass(KeyedObject): def __init__(self,key='dummy',attr1='a1',attr2='a2'): KeyedObject.__init__(self,key=key) self.attr1 = attr1 self.attr2 = attr2 set_of_keyed_objects = SetOfKeyedObjects(TestLoadClass) set_of_keyed_objects.__load__(base_dir=os.path.join(os.getcwd(),'mockdb/test_data'),no_children=True) d = set_of_keyed_objects.__retrieve_values_by_key__('attr1')
def test_load_set(): class TestLoadClass(KeyedObject): def __init__(self,key='dummy',attr1='a1',attr2='a2'): KeyedObject.__init__(self,key=key) self.attr1 = attr1 self.attr2 = attr2 set_of_keyed_objects = SetOfKeyedObjects(TestLoadClass) set_of_keyed_objects.__load__(base_dir=os.path.join(os.getcwd(),'mockdb/test_data'),no_children=True,key=1) try: if set_of_keyed_objects.objects['1'].attr1 != 'attribute1': print 'load did not load the correct attributes.' except KeyError: print 'Specifying the key in load did not work.' try: if set_of_keyed_objects.objects['2'].attr1 != 'attribute1': "More keys than specified were loaded." except KeyError: pass