88 lines
2.9 KiB
Python
88 lines
2.9 KiB
Python
|
import unittest
|
||
|
|
|
||
|
|
from rapidfuzz import process, fuzz, utils
|
||
|
|
|
||
|
|
class RatioTest(unittest.TestCase):
|
||
|
|
def setUp(self):
|
||
|
|
self.s1 = "new york mets"
|
||
|
|
self.s1a = "new york mets"
|
||
|
|
self.s2 = "new YORK mets"
|
||
|
|
self.s3 = "the wonderful new york mets"
|
||
|
|
self.s4 = "new york mets vs atlanta braves"
|
||
|
|
self.s5 = "atlanta braves vs new york mets"
|
||
|
|
self.s6 = "new york mets - atlanta braves"
|
||
|
|
|
||
|
|
def testEqual(self):
|
||
|
|
self.assertEqual(fuzz.ratio(self.s1, self.s1a),100)
|
||
|
|
|
||
|
|
def testCaseInsensitive(self):
|
||
|
|
self.assertNotEqual(fuzz.ratio(self.s1, self.s2),100)
|
||
|
|
self.assertEqual(fuzz.ratio(self.s1, self.s2, processor=True),100)
|
||
|
|
|
||
|
|
def testPartialRatio(self):
|
||
|
|
self.assertEqual(fuzz.partial_ratio(self.s1, self.s3),100)
|
||
|
|
|
||
|
|
def testTokenSortRatio(self):
|
||
|
|
self.assertEqual(fuzz.token_sort_ratio(self.s1, self.s1a),100)
|
||
|
|
|
||
|
|
def testPartialTokenSortRatio(self):
|
||
|
|
self.assertEqual(fuzz.partial_token_sort_ratio(self.s1, self.s1a),100)
|
||
|
|
self.assertEqual(fuzz.partial_token_sort_ratio(self.s4, self.s5),100)
|
||
|
|
|
||
|
|
def testTokenSetRatio(self):
|
||
|
|
self.assertEqual(fuzz.token_set_ratio(self.s4, self.s5),100)
|
||
|
|
|
||
|
|
def testPartialTokenSetRatio(self):
|
||
|
|
self.assertEqual(fuzz.partial_token_set_ratio(self.s4, self.s5),100)
|
||
|
|
|
||
|
|
def testQuickRatioEqual(self):
|
||
|
|
self.assertEqual(fuzz.QRatio(self.s1, self.s1a), 100)
|
||
|
|
|
||
|
|
def testQuickRatioCaseInsensitive(self):
|
||
|
|
self.assertEqual(fuzz.QRatio(self.s1, self.s2), 100)
|
||
|
|
|
||
|
|
def testQuickRatioNotEqual(self):
|
||
|
|
self.assertNotEqual(fuzz.QRatio(self.s1, self.s3), 100)
|
||
|
|
|
||
|
|
def testWRatioEqual(self):
|
||
|
|
self.assertEqual(fuzz.WRatio(self.s1, self.s1a), 100)
|
||
|
|
|
||
|
|
def testWRatioCaseInsensitive(self):
|
||
|
|
self.assertEqual(fuzz.WRatio(self.s1, self.s2), 100)
|
||
|
|
|
||
|
|
def testWRatioPartialMatch(self):
|
||
|
|
# a partial match is scaled by .9
|
||
|
|
self.assertEqual(fuzz.WRatio(self.s1, self.s3), 90)
|
||
|
|
|
||
|
|
def testWRatioMisorderedMatch(self):
|
||
|
|
# misordered full matches are scaled by .95
|
||
|
|
self.assertEqual(fuzz.WRatio(self.s4, self.s5), 95)
|
||
|
|
|
||
|
|
def testWRatioUnicode(self):
|
||
|
|
self.assertEqual(fuzz.WRatio(self.s1, self.s1a), 100)
|
||
|
|
|
||
|
|
def testQRatioUnicode(self):
|
||
|
|
self.assertEqual(fuzz.WRatio(self.s1, self.s1a), 100)
|
||
|
|
|
||
|
|
def testEmptyStrings(self):
|
||
|
|
self.assertEqual(fuzz.ratio("", ""), 100)
|
||
|
|
self.assertEqual(fuzz.partial_ratio("", ""), 100)
|
||
|
|
|
||
|
|
def testNoneString(self):
|
||
|
|
self.assertEqual(fuzz.ratio("", None), 0)
|
||
|
|
self.assertEqual(fuzz.partial_ratio("", None), 0)
|
||
|
|
|
||
|
|
def testWRatioUnicodeString(self):
|
||
|
|
s1 = "Á"
|
||
|
|
s2 = "ABCD"
|
||
|
|
score = fuzz.WRatio(s1, s2)
|
||
|
|
self.assertEqual(0, score)
|
||
|
|
|
||
|
|
def testQRatioUnicodeString(self):
|
||
|
|
s1 = "Á"
|
||
|
|
s2 = "ABCD"
|
||
|
|
score = fuzz.QRatio(s1, s2)
|
||
|
|
self.assertEqual(0, score)
|
||
|
|
|
||
|
|
if __name__ == '__main__':
|
||
|
|
unittest.main()
|