def test_get_at(self): n = Node(1) ll = Linkedlist() ll.insert_at(0, n) self.assertEqual(ll.get_at(0), n) n = Node(2) ll.insert_at(10, n) self.assertEqual(ll.get_at(1), n)
def test_iter(self): ll = Linkedlist() n1 = Node(1) ll.insert_at(0, n1) n2 = Node(2) ll.insert_at(10, n2) n3 = Node(3) ll.insert_at(1, n3) for node in ll: print node.data
def test_remove_at(self): n1 = Node(1) ll = Linkedlist() self.assertEqual(ll.remove_at(0), None) ll.insert_at(0, n1) n2 = Node(2) ll.insert_at(10, n2) n3 = Node(3) ll.insert_at(1, n3) self.assertEqual(ll.remove_at(2), n3) self.assertEqual(ll.remove_at(0), n1)