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