def test_get_attribute(self): t = Tier(id='t', attributes={'one': 1, 'two': 2}) igt = Igt(tiers=[t], attributes={'three': 3}) self.assertEqual(t.get_attribute('one'), 1) self.assertEqual(t.get_attribute('two'), 2) self.assertIs(t.get_attribute('three'), None) self.assertEqual(t.get_attribute('three', inherit=True), 3) self.assertEqual(t.get_attribute('three', default=4), 4)
def test_get_attribute(self): t = Tier(id='t', attributes={'one': 1, 'two': 2}) igt = Igt(tiers=[t], attributes={'three': 3}) assert t.get_attribute('one') == 1 assert t.get_attribute('two') == 2 assert t.get_attribute('three') is None assert t.get_attribute('three', inherit=True) == 3 assert t.get_attribute('three', default=4) == 4