def testClearShouldRemoveEveryting(self): t = Trie() t.add_word('python', 1) t.add_word('ada', 2) t.add_word('perl', 3) t.add_word('pascal', 4) t.add_word('php', 5) self.assertEqual(len(t), 5) self.assertEqual(len(list(t.items())), 5) t.clear() self.assertEqual(len(t), 0) self.assertEqual(len(list(t.items())), 0)
def testItemsShouldReturnAllItemsAlreadyAddedToTheTrie(self): t = Trie() t.add_word('python', 1) t.add_word('ada', 2) t.add_word('perl', 3) t.add_word('pascal', 4) t.add_word('php', 5) result = list(t.items()) self.assertEquals(len(result), 5) self.assertIn(('python', 1), result) self.assertIn(('ada', 2), result) self.assertIn(('perl', 3), result) self.assertIn(('pascal', 4), result) self.assertIn(('php', 5), result)