Sprint-Challenge--Data-Structures-Python/reverse/test_reverse.py at 8dcaa94877bdb5c78b80810407f86db4a3adb923 · bloominstituteoftechnology/Sprint-Challenge--Data-Structures-Python · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
import unittest
from reverse import LinkedList


class LinkedListTests(unittest.TestCase):
    def setUp(self):
        self.list = LinkedList()

    def test_add_to_head(self):
        self.list.add_to_head(1)
        self.assertEqual(self.list.head.value, 1)
        self.list.add_to_head(2)
        self.assertEqual(self.list.head.value, 2)

    def test_contains(self):
        self.list.add_to_head(1)
        self.list.add_to_head(2)
        self.list.add_to_head(10)
        self.assertTrue(self.list.contains(2))
        self.assertTrue(self.list.contains(10))
        self.assertFalse(self.list.contains(1000))

    def test_empty_reverse(self):
        self.list.reverse_list()
        self.assertEqual(self.list.head, None)

    def test_single_reverse(self):
        self.list.add_to_head(1)
        self.list.reverse_list()
        self.assertEqual(self.list.head.value, 1)

    def test_longer_reverse(self):
        self.list.add_to_head(1)
        self.list.add_to_head(2)
        self.list.add_to_head(3)
        self.list.add_to_head(4)
        self.list.add_to_head(5)
        self.assertEqual(self.list.head.value, 5)
        self.list.reverse_list()
        self.assertEqual(self.list.head.value, 1)
        self.assertEqual(self.list.head.get_next().value, 2)
        self.assertEqual(self.list.head.get_next().get_next().value, 3)


if __name__ == '__main__':
    unittest.main()