def test_behavior(self): d = AttributeDict(something="foobar", another_thing="haha") self.assertEqual(d.something, "foobar") self.assertEqual(d.another_thing, "haha") d = attrdict(something="foobar", another_thing="haha") self.assertEqual(d.something, "foobar") self.assertEqual(d.another_thing, "haha")
def test_AttributeError_when_key_not_found(self): d = AttributeDict(something="foobar", another_thing="haha") a = attrdict(something="foobar", another_thing="haha") def foo_wrapper(d): return d.not_present self.assertRaises(AttributeError, foo_wrapper, d) self.assertRaises(AttributeError, foo_wrapper, a)