def test_items(self): ht = TreeMap() assert ht.items() == [] ht.set('I', 1) assert ht.items() == [('I', 1)] ht.set('V', 5) self.assertCountEqual(ht.items(), [('I', 1), ('V', 5)]) ht.set('X', 10) self.assertCountEqual(ht.items(), [('I', 1), ('V', 5), ('X', 10)])
def test_set(self): tm = TreeMap() assert tm.size == 0 tm.set(1, 2) assert tm.size == 1 tm.set(2, 3) assert tm.size == 2 tm.set(3, 4) assert tm.size == 3 self.assertCountEqual(tm.items(), [(1, 2), (2, 3), (3, 4)]) tm.set(3, 5) assert tm.size == 3 self.assertCountEqual(tm.items(), [(1, 2), (2, 3), (3, 5)])
def test_delete(self): tm = TreeMap() assert tm.size == 0 tm.set(1, 2) assert tm.size == 1 tm.set(2, 3) assert tm.size == 2 tm.set(3, 4) assert tm.size == 3 self.assertCountEqual(tm.items(), [(1, 2), (2, 3), (3, 4)]) tm.delete(3) assert tm.size == 2 self.assertCountEqual(tm.items(), [(1, 2), (2, 3)]) tm.delete(2) assert tm.size == 1 tm.delete(1) assert tm.size == 0
def test_items(self): tm = TreeMap({"c": "C", "b": "B", "a": "A"}) keys = [k for k, v in tm.items()] self.assertSequenceEqual(["a", "b", "c"], keys)
def test_items(self): tree = TreeMap([(2, 'B'), (1, 'A'), (3, 'C')]) assert tree.items() == [(1, 'A'), (2, 'B'), (3, 'C')]
def test_init_with_list(self): tm = TreeMap([(1, 2), (2, 3), (3, 4)]) assert tm.size == 3 self.assertCountEqual(tm.items(), [(1, 2), (2, 3), (3, 4)])