Python Node.getNext 예제들

예제 #1

파일: 2.1_Remove_Dups.py 프로젝트: nairihar/CtCI-6th-Edition-Python

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)

예제 #2

파일: 2.3_Delete_Middle_Node.py 프로젝트: nairihar/CtCI-6th-Edition-Python

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)