def test_om_getattr_dict(self): element = { 'one': 1, 'none': None } self.assertEqual(utils.om_getattr(element, 'one'), element['one']) self.assertEqual(utils.om_getattr(element, 'none'), element['none'])
def test_om_getattr_object(self): class Test: def __init__(self): self.one = 1 self.none = None element = Test() self.assertEqual(utils.om_getattr(element, 'one'), element.one) self.assertEqual(utils.om_getattr(element, 'none'), element.none)