forked from benevolentprof/inf1340_2015_asst2
-
Notifications
You must be signed in to change notification settings - Fork 0
/
test_exercise3.py
136 lines (98 loc) · 3.33 KB
/
test_exercise3.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
125
126
127
128
129
130
131
132
133
134
135
#!/usr/bin/env python
""" Assignment 2, Exercise 3, INF1340, Fall, 2015. DBMS
Test module for exercise3.py
"""
__author__ = 'Susan Sim'
__email__ = "ses@drsusansim.org"
__copyright__ = "2015 Susan Sim"
__license__ = "MIT License"
from exercise3 import union, intersection, difference, MismatchedAttributesException
###########
# TABLES ##
###########
GRADUATES = [["Number", "Surname", "Age"],
[7274, "Robinson", 37],
[7432, "O'Malley", 39],
[9824, "Darkes", 38]]
MANAGERS = [["Number", "Surname", "Age"],
[9297, "O'Malley", 56],
[7432, "O'Malley", 39],
[9824, "Darkes", 38]]
STAFF = [["ID", "Name", "Position", "Age"],
[7466, "Frank Underwood", "President", 55],
[5323, "Jeffrey Winger", "Former Lawyer", 40],
[1233, "Piper Chapman", "Prisoner", 31],
[1233, "Piper Chapman", "Prisoner", 31],
[4383, "Oliver Dart", "Unknown", 23]]
PROFESSORS = [["Number", "Surname", "Age"],
[7434, "Silva", 33],
[8374, "Cray", 40],
[9824, "Darkes", 38]]
BLANK = [["Number", "Surname", "Age"]]
#####################
# HELPER FUNCTIONS ##
#####################
def is_equal(t1, t2):
return sorted(t1) == sorted(t2)
###################
# TEST FUNCTIONS ##
###################
def test_union():
"""
Test union operation.
"""
result = [["Number", "Surname", "Age"],
[9297, "O'Malley", 56],
[7432, "O'Malley", 39],
[9824, "Darkes", 38],
[7274, "Robinson", 37]]
assert result == union(MANAGERS, GRADUATES)
result = [["Number", "Surname", "Age"],
[7434, "Silva", 33],
[8374, "Cray", 40],
[9824, "Darkes", 38],
[7274, "Robinson", 37],
[7432, "O'Malley", 39]]
assert result == union(PROFESSORS, GRADUATES)
result = [['Number', 'Surname', 'Age'],
[7274, 'Robinson', 37],
[7432, "O'Malley", 39],
[9824, 'Darkes', 38],
[9297, "O'Malley", 56]]
assert result == union(GRADUATES,MANAGERS)
try:
assert result == union(GRADUATES, STAFF)
except MismatchedAttributesException:
assert True
def test_intersection():
"""
Test intersection operation.
"""
result = [["Number", "Surname", "Age"],
[7432, "O'Malley", 39],
[9824, "Darkes", 38]]
assert result == intersection(GRADUATES, MANAGERS)
result = [["Number", "Surname", "Age"],
[9824, "Darkes", 38]]
assert result == intersection(PROFESSORS, MANAGERS)
assert intersection(GRADUATES, BLANK) is None
try:
assert result == intersection(GRADUATES, STAFF)
except MismatchedAttributesException:
assert True
def test_difference():
"""
Test difference operation.
"""
result = [["Number", "Surname", "Age"],
[7274, "Robinson", 37]]
assert result == difference(GRADUATES, MANAGERS)
result = [["Number", "Surname", "Age"],
[9297, "O'Malley", 56],
[7432, "O'Malley", 39]]
assert result == difference(MANAGERS, PROFESSORS)
assert difference(BLANK, GRADUATES) is None
try:
assert result == difference(GRADUATES, STAFF)
except MismatchedAttributesException:
assert True