File tests.py of Package python-pylev

import itertools
import unittest
import pylev

test_data = [
    ('classic', "kitten", "sitting", 3),
    ('same', "kitten", "kitten", 0),
    ('empty', "", "", 0),
    ('a', "meilenstein", "levenshtein", 4),
    ('b', "levenshtein", "frankenstein", 6),
    ('c', "confide", "deceit", 6),
    ('d', "CUNsperrICY", "conspiracy", 8),
]

test_functions = [
    # pylev.classic_levenshtein,   # disabled because it is so slow
    pylev.recursive_levenshtein,
    pylev.wf_levenshtein,
    pylev.wfi_levenshtein
]

class Tests(unittest.TestCase):

    def test_damerau_levenshtein(seld):
        assert pylev.damerau_levenshtein("ba", "abc") == 2
        assert pylev.damerau_levenshtein("foobar", "foobra") == 1
        assert pylev.damerau_levenshtein("fee", "deed") == 2


def _mk_test_fn(fn, a, b, expected):
    def _test_fn(self):
        self.assertEqual(fn(a, b), expected)
        self.assertEqual(fn(b, a), expected)

    return _test_fn


for lev_fn, data in itertools.product(test_functions, test_data):
    name, a, b, expected = data
    test_fn = _mk_test_fn(lev_fn, a, b, expected)
    setattr(Tests, "test_%s_%s" % (name, lev_fn.__name__), test_fn)


if __name__ == '__main__':
    unittest.main()