Ejemplo n.º 1
0
    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)
Ejemplo n.º 2
0
 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)