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
|
from unittest import TestCase
import fingerprints
from fingerprints import fingerprint as fp
class FingerprintsTest(TestCase):
def test_normal_names(self):
self.assertEqual(fp("Mr. Boaty McBoatface"), "boaty mcboatface")
self.assertEqual(fp("Open S.A.R.L."), "open sarl")
self.assertEqual(fp("Johnson's Coffee Shop"), "coffee johnsons shop")
self.assertEqual(fp("New York, New York"), "new york")
def test_replacers(self):
self.assertEqual(fp("Foo Limited"), "foo ltd")
self.assertEqual(
fp("Foo International bla Limited"), "bla foo intl ltd"
) # noqa
self.assertEqual(fp("Foo International Limited"), "foo intl ltd")
def test_cyrillic(self):
self.assertEqual(fp("РАДИК ІВАН ЛЬВОВИЧ"), "ivan lvovic radik")
self.assertEqual(
fp("КУШНАРЬОВ ДМИТРО ВІТАЛІЙОВИЧ"), "dmitro kusnarov vitalijovic"
) # noqa
self.assertEqual(
fp("Порошенко Петро Олексійович"), "oleksijovic petro porosenko"
) # noqa
def test_turcic(self):
self.assertEqual(fp("FUAD ALIYEV ƏHMƏD OĞLU"), "ahmad aliyev fuad oglu") # noqa
def test_german(self):
self.assertEqual(fp("Siemens Aktiengesellschaft"), "ag siemens") # noqa
self.assertEqual(
fp("Software und- Systemgesellschaft mit beschr Haftung"), # noqa
"gmbh software systemgesellschaft und",
) # noqa
def test_company(self):
self.assertEqual(fp('S.R.L. "Magic-Arrow" ICS'), "arrow ics magic srl") # noqa
def test_brackets(self):
self.assertEqual(fp("Foo (Bar) CORPORATION"), "corp foo") # noqa
def test_remove(self):
rem = fingerprints.remove_types("Siemens Aktiengesellschaft")
self.assertEqual(rem, "siemens") # noqa
rem = fingerprints.remove_types("Siemens AG")
self.assertEqual(rem, "siemens") # noqa
rem = fingerprints.remove_types("Foo Limited")
self.assertEqual(rem, "foo")
|
