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())