-
Notifications
You must be signed in to change notification settings - Fork 0
/
etl_test.py
68 lines (59 loc) · 1.7 KB
/
etl_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
import unittest
from etl import (
transform,
)
# Tests adapted from `problem-specifications//canonical-data.json`
class EtlTest(unittest.TestCase):
def test_single_letter(self):
legacy_data = {1: ["A"]}
data = {"a": 1}
self.assertEqual(transform(legacy_data), data)
def test_single_score_with_multiple_letters(self):
legacy_data = {1: ["A", "E", "I", "O", "U"]}
data = {"a": 1, "e": 1, "i": 1, "o": 1, "u": 1}
self.assertEqual(transform(legacy_data), data)
def test_multiple_scores_with_multiple_letters(self):
legacy_data = {1: ["A", "E"], 2: ["D", "G"]}
data = {"a": 1, "d": 2, "e": 1, "g": 2}
self.assertEqual(transform(legacy_data), data)
def test_multiple_scores_with_differing_numbers_of_letters(self):
legacy_data = {
1: ["A", "E", "I", "O", "U", "L", "N", "R", "S", "T"],
2: ["D", "G"],
3: ["B", "C", "M", "P"],
4: ["F", "H", "V", "W", "Y"],
5: ["K"],
8: ["J", "X"],
10: ["Q", "Z"],
}
data = {
"a": 1,
"b": 3,
"c": 3,
"d": 2,
"e": 1,
"f": 4,
"g": 2,
"h": 4,
"i": 1,
"j": 8,
"k": 5,
"l": 1,
"m": 3,
"n": 1,
"o": 1,
"p": 3,
"q": 10,
"r": 1,
"s": 1,
"t": 1,
"u": 1,
"v": 4,
"w": 4,
"x": 8,
"y": 4,
"z": 10,
}
self.assertEqual(transform(legacy_data), data)
if __name__ == "__main__":
unittest.main()