forked from mpchoy/python-skiplist
-
Notifications
You must be signed in to change notification settings - Fork 0
/
sortedlist_test.py
executable file
·124 lines (101 loc) · 3.23 KB
/
sortedlist_test.py
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
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
import unittest
from sortedlist import Node
from sortedlist import SortedList
from random import shuffle
class TestSortedListFunctions(unittest.TestCase):
def test_init_node_number(self):
node = Node(5)
self.assertEqual(node.data, 5)
self.assertIsNone(node.next)
def test_init_node_string(self):
string = 'yet a string'
node = Node(string)
self.assertEqual(node.data, string)
def test_init_SL(self):
sl = SortedList()
self.assertIsNone(sl.head)
self.assertIsNone(sl.tail)
def test_insert_first(self):
sl = SortedList()
sl.insert( 1 )
self.assertEqual(sl.head.data, 1)
self.assertEqual(sl.tail.data, 1)
def test_insert_second(self):
sl = SortedList()
sl.insert( 2 )
sl.insert( 1 )
self.assertEqual(sl.head.data, 1)
self.assertEqual(sl.head.next.data, 2)
self.assertEqual(sl.tail.data, 2)
self.assertIsNone(sl.tail.next)
def test_insert_second_1(self):
sl = SortedList()
sl.insert( 1 )
sl.insert( 2 )
self.assertEqual(sl.head.data, 1)
self.assertEqual(sl.head.next.data, 2)
self.assertEqual(sl.tail.data, 2)
self.assertIsNone(sl.tail.next)
def test_insert_third(self):
sl = SortedList()
sl.insert( 3 )
sl.insert( 2 )
sl.insert( 1 )
self.assertEqual(sl.head.data, 1)
self.assertEqual(sl.head.next.data, 2)
self.assertEqual(sl.head.next.next.data, 3)
self.assertEqual(sl.tail.data, 3)
self.assertIsNone(sl.tail.next)
def test_insert_third_1(self):
sl = SortedList()
sl.insert( 1 )
sl.insert( 2 )
sl.insert( 3 )
self.assertEqual(sl.head.data, 1)
self.assertEqual(sl.head.next.data, 2)
self.assertEqual(sl.head.next.next.data, 3)
self.assertEqual(sl.tail.data, 3)
self.assertIsNone(sl.tail.next)
def test_remove(self):
sl = SortedList()
sl.insert( 1 )
sl.remove( 1 )
sl.remove( 1 )
self.assertIsNone(sl.head)
self.assertIsNone(sl.tail)
def test_remove_first(self):
sl = SortedList()
sl.insert( 2 )
sl.insert( 1 )
sl.remove( 1 )
self.assertEqual(sl.head.data, 2)
self.assertEqual(sl.tail.data, 2)
self.assertIsNone(sl.tail.next)
def test_remove_last(self):
sl = SortedList()
sl.insert( 2 )
sl.insert( 1 )
sl.remove( 2 )
self.assertEqual(sl.head.data, 1)
self.assertEqual(sl.tail.data, 1)
self.assertIsNone(sl.tail.next)
def test_remove_middle(self):
sl = SortedList()
sl.insert( 3 )
sl.insert( 2 )
sl.insert( 1 )
sl.remove( 2 )
self.assertEqual(sl.head.data, 1)
self.assertEqual(sl.head.next.data, 3)
self.assertEqual(sl.tail.data, 3)
self.assertIsNone(sl.tail.next)
def test_stress(self):
sl = SortedList()
items = list( range(3000) )
shuffle( items )
for i in range(3000):
sl.insert( items[i] )
sl.remove( i / 2 )
sl.remove( i / 3 )
if __name__ == '__main__':
unittest.main()