def test_update(self): d = test_items_seq() tree = AnderssonTree() tree.update(d) self.assertEqual(len(tree), len(d)) self.assertEqual(set(tree.keys()), set(x[0] for x in d)) self.assertEqual(set(tree.values()), set(x[1] for x in d)) self.check_aatree_properties(tree)
def test_pickle_protocol(self): s = test_items_seq() tree1 = AnderssonTree(s) pickle_str = pickle.dumps(tree1, -1) tree2 = pickle.loads(pickle_str) self.assertEqual(len(tree1), len(tree2)) self.assertEqual(list(tree1.keys()), list(tree2.keys())) self.assertEqual(list(tree1.values()), list(tree2.values())) self.check_aatree_properties(tree1) self.check_aatree_properties(tree2)
def test_values(self): s = test_items_seq() tree = AnderssonTree(s) values = set(x[1] for x in s) self.assertEqual(set(tree.values()), values)