class NodeFunctionsTest(unittest.TestCase):
def setUp(self):
self.n1 = Node(1)
self.n2 = Node(2)
self.n3 = Node(3)
self.n3.next = self.n2
def test_get_data(self):
self.assertEqual(self.n1.get_data(), 1)
self.assertEqual(self.n3.get_data(), 3)
def test_get_next(self):
self.assertEqual(self.n1.get_next(), None)
self.assertEqual(self.n3.get_next(), self.n2)
def test_set_data(self):
self.n1.set_data(11)
self.assertEqual(self.n1.data, 11)
def test_set_next(self):
self.n2.set_next(self.n1)
self.assertEqual(self.n2.next, self.n1)
