~linuxgoose/linguistics-robin

linguistics-robin/tests/test_corner_cases.py -rw-r--r-- 2.8 KiB
6ef2ba53 — Jordan Robinson Refactor GitHub Actions workflow for publishing a month ago 
                                                                                
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

import pytest
from linguistics_robin import Soundex, RefinedSoundex, FuzzySoundex, NYSIIS, \
    Metaphone, DoubleMetaphone, Caverphone, Caverphone2
from linguistics_robin.exceptions import EmptyStringError

def test_caverphone():
    caverphone = Caverphone()

    assert caverphone.phonetics('maurice') == 'MRSA'
    assert caverphone.phonetics('bob') == 'PP'
    assert caverphone.phonetics('walter') == 'WTA'
    assert caverphone.phonetics('Maclaverty') == 'MKLFTA'

    with pytest.raises(EmptyStringError):
        caverphone.phonetics('')

def test_caverphone2():
    caverphone = Caverphone2()

    assert caverphone.phonetics('Thompson') == 'TMPSN11111'
    assert caverphone.phonetics('Lee') == 'LA11111111'
    assert caverphone.phonetics('Stevenson') == 'STFNSN1111'
    assert caverphone.phonetics('Peter') == 'PTA1111111'

    with pytest.raises(EmptyStringError):
        caverphone.phonetics('')

def test_doublemetaphone():
    dm = DoubleMetaphone()

    assert dm.phonetics('maurice') == ('MRS', None)
    assert dm.phonetics('bob') == ('PP', None)
    assert dm.phonetics('walter') == ('ALTR', 'FLTR')

    with pytest.raises(EmptyStringError):
        dm.phonetics('')

def test_metaphone():
    metaphone = Metaphone()

    assert metaphone.phonetics('maurice') == 'MRS'
    assert metaphone.phonetics('bob') == 'BB'

    with pytest.raises(EmptyStringError):
        metaphone.phonetics('')

def test_nysiis():
    nysiis = NYSIIS()

    assert nysiis.phonetics('bob') == 'BAB'
    assert nysiis.phonetics('aa') == 'A'
    assert nysiis.phonetics('b') == 'B'
    assert nysiis.phonetics('cat') == 'CAT'
    assert nysiis.phonetics('s') == 'S'
    assert nysiis.phonetics('w') == 'W'

    with pytest.raises(EmptyStringError):
        nysiis.phonetics('')

def test_soundex():
    soundex = Soundex()
    
    assert soundex.phonetics('h') == 'H000'
    assert soundex.phonetics('hh') == 'H000'
    assert soundex.phonetics('hhh') == 'H000'
    assert soundex.phonetics('d') == 'D000'
    assert soundex.phonetics('dd') == 'D000'
    assert soundex.phonetics('ddd') == 'D000'
    assert soundex.phonetics('ddm') == 'D500'
    assert soundex.phonetics('ddmmmm') == 'D500'
    assert soundex.phonetics('Pffister') == 'P236'
    assert soundex.phonetics('Pfister') == 'P236'

    with pytest.raises(EmptyStringError):
        soundex.phonetics('')


def test_refined_soundex():
    soundex = RefinedSoundex()

    assert soundex.phonetics('h') == 'H'
    assert soundex.phonetics('d') == 'D6'

    with pytest.raises(EmptyStringError):
        soundex.phonetics('')


def test_fuzzy_soundex():
    soundex = FuzzySoundex()

    assert soundex.phonetics('Catharine') == 'K365'
    assert soundex.phonetics('Katharine') == 'K365'

    with pytest.raises(EmptyStringError):
        soundex.phonetics('')