def get_next(self): """Test get_next method.""" n1 = Node(1) n2 = Node(2) # by default, next is None self.assertIsNone(n1.next) n1.set_next(n2) self.assertEqual(n1.get_next(), n2)
def set_next(self): """Test set_next method.""" n1 = Node(1) n2 = Node(2) # by default, next is None self.assertIsNone(n1.next) n1.set_next(n2) # data from next is data from node 2 self.assertEqual(n1.get_next().get_data(), 2)