-
Notifications
You must be signed in to change notification settings - Fork 19
/
groupmap.py
56 lines (43 loc) · 1.3 KB
/
groupmap.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
import json, urllib2
class Translator:
map = {}
def __init__(self, sourceportal):
filename = "../kategorien/" + sourceportal + "2deutschland.json"
self.map = json.loads( open(filename, 'r').read())
def translate(self, categories=[]):
if not isinstance(categories, list):
categories = [categories]
out = []
if not categories:
return out
for cat in categories:
if cat in self.map.keys():
out = out + self.map[cat]
return out
import unittest
class TestMapping(unittest.TestCase):
@classmethod
def setUpClass(cls):
cls.u = Translator('berlin')
def test_b2d(self):
translated = self.u.translate(['verkehr','wahl'])
print translated
self.assertEqual(translated, ['transport_verkehr','politik_wahlen'])
def test_atom(self):
translated = self.u.translate('verkehr')
print translated
self.assertEqual(translated, ['transport_verkehr'])
def test_missing(self):
translated = self.u.translate(['verkehr','xyz','wahl'])
print translated
self.assertEqual(translated, ['transport_verkehr','politik_wahlen'])
def test_empty(self):
translated = self.u.translate()
print translated
self.assertEqual(translated, [])
def test_none(self):
translated = self.u.translate(None)
print translated
self.assertEqual(translated, [])
if __name__ == '__main__':
unittest.main()