def test_find_node_by_value(self): node1 = LinkedListNode("node 1") node2 = LinkedListNode("node 2") node3 = LinkedListNode("node 3") my_linked_list = LinkedList() my_linked_list.add_first(node1) my_linked_list.add_node_after(node1, node2) my_linked_list.add_last(node3) node_to_find = my_linked_list.find_node_by_value("node 1") self.assertEqual(node_to_find, node1) node_to_find = my_linked_list.find_node_by_value("node 2") self.assertEqual(node_to_find, node2) node_to_find = my_linked_list.find_node_by_value("node 3") self.assertEqual(node_to_find, node3)
def test_add_node_after(self): node1 = LinkedListNode("node 1") node2 = LinkedListNode("node 2") node3 = LinkedListNode("node 3") node4 = LinkedListNode("node 4") my_linked_list = LinkedList() my_linked_list.add_first(node1) my_linked_list.add_node_after(node1, node2) self.assertEqual(my_linked_list.head.node_value, 'node 1') self.assertEqual( my_linked_list.get_node_by_index(1).node_value, 'node 2') self.assertEqual(my_linked_list.length, 2) with self.assertRaises(ValueError): my_linked_list.add_node_after(node3, node4) my_linked_list.add_node_after(node2, node3) self.assertEqual( my_linked_list.get_node_by_index(1).node_value, 'node 2') self.assertEqual( my_linked_list.get_node_by_index(2).node_value, 'node 3') self.assertEqual(my_linked_list.tail.node_value, 'node 3') self.assertEqual(my_linked_list.length, 3) print(my_linked_list)