/
tests.py
78 lines (63 loc) · 1.88 KB
/
tests.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
import string
import random
import unittest
from sortedlist import SortedList
random.seed(0)
class SortedListTestCase(unittest.TestCase):
def setUp(self):
self.sorted_list = SortedList()
@property
def is_sorted(self):
for i, item in enumerate(self.sorted_list):
if i == 0:
current = item
else:
if current > item:
return False
return True
def test_append_is_sorted(self):
items = []
for _ in xrange(200):
num = random.uniform(-100, 100)
self.sorted_list.append(num)
self.assertTrue(self.is_sorted)
def test_append_keeps_items(self):
items = []
for _ in xrange(200):
num = random.uniform(-100, 100)
self.sorted_list.append(num)
items.append(num)
items.sort()
for i, j in zip(items, self.sorted_list):
self.assertEqual(i, j)
def test_extend_is_sorted(self):
nums = range(-100, 100)
random.shuffle(nums)
self.sorted_list.extend(nums)
self.assertTrue(self.is_sorted)
def test_extend_keeps_items(self):
nums = range(-100, 100)
random.shuffle(nums)
self.sorted_list.extend(nums)
nums.sort()
for i, j in zip(nums, self.sorted_list):
self.assertEqual(i, j)
def test_setitem_is_sorted(self):
nums = xrange(-100, 100)
self.sorted_list.extend(nums)
for _ in xrange(200):
self.sorted_list[random.randint(0, len(self.sorted_list) - 1)] = random.uniform(-100, 100)
self.assertTrue(self.is_sorted)
def test_setitem_keeps_items(self):
nums = range(-100, 100)
self.sorted_list.extend(nums)
for _ in xrange(200):
index = random.randint(0, len(self.sorted_list) - 1)
num = random.uniform(-100, 100)
self.sorted_list[index] = num
nums[index] = num
nums.sort()
for i, j in zip(self.sorted_list, nums):
self.assertEqual(i, j)
if __name__ == "__main__":
unittest.main()