class Test(unittest.TestCase): '''Test Cases''' def setUp(self): self.node = Node(1) self.node.connect(2).connect(3).connect(2).connect(4) removeDups(self.node) def test_node_values(self): self.assertEqual(self.node.getValue(), 1) self.assertEqual(self.node.getNext().getValue(), 2) self.assertEqual(self.node.getNext().getNext().getValue(), 3) self.assertEqual(self.node.getNext().getNext().getNext().getValue(), 4) self.assertEqual(self.node.getNext().getNext().getNext().getNext(), None)
class Test(unittest.TestCase): '''Test Cases''' def setUp(self): self.node = Node(1) self.node.connect(2).connect(3).connect(4).connect(5) deleteMiddleNode(self.node) self.node2 = Node(1) self.node2.connect(2).connect(3).connect(4) deleteMiddleNode(self.node2) def test_node_values_1(self): self.assertEqual(self.node.getValue(), 1) self.assertEqual(self.node.getNext().getValue(), 2) self.assertEqual(self.node.getNext().getNext().getValue(), 4) self.assertEqual(self.node.getNext().getNext().getNext().getValue(), 5) self.assertEqual(self.node.getNext().getNext().getNext().getNext(), None) def test_node_values_2(self): self.assertEqual(self.node2.getValue(), 1) self.assertEqual(self.node2.getNext().getValue(), 2) self.assertEqual(self.node2.getNext().getNext().getValue(), 4) self.assertEqual(self.node2.getNext().getNext().getNext(), None)