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)
