def test_keys_items_and_values(self):
d = AttributeDict(a="ah", b="bh", c="ch")
self.assertEqual(set(d.keys()), set(["a", "b", "c"]))
self.assertEqual(set(d.values()), set(["ah", "bh", "ch"]))
self.assertEqual(set(d.items()),
set([("a", "ah"), ("b", "bh"), ("c", "ch")]))
def test_get(self):
d = AttributeDict(a="ah")
self.assertEqual(d.get("a"), "ah")
self.assertEqual(d.get("b"), None)
self.assertEqual(d.get("b", "foo"), "foo")
def test_items(self):
d = AttributeDict(a="a", b="b", c="c")
another = AttributeDict()
for k, v in d.items():
another[k] = v
self.assertDictEqual(d, another)
def test_setattr(self): d = AttributeDict() d.something = "foobar" self.assertEqual(d.something, "foobar")
def test_update(self): d = AttributeDict(a="b") d.update(AttributeDict(b="c")) self.assertDictEqual(d, dict(a="b", b="c"))
def test_fromkeys(self): d = AttributeDict(a="a", b="b", c="c") another = AttributeDict.fromkeys(d, "Hmm") self.assertDictEqual(another, dict(a="Hmm", b="Hmm", c="Hmm"))
def test_clear(self): d = AttributeDict(a="ah", b="foo") d.clear() self.assertDictEqual(d, dict())
