diff --git a/test/test_similarity.py b/test/test_similarity.py index 573447d62..905db973e 100644 --- a/test/test_similarity.py +++ b/test/test_similarity.py @@ -2,7 +2,10 @@ from test.picardtestcase import PicardTestCase -from picard.similarity import similarity +from picard.similarity import ( + similarity, + similarity2, +) class SimilarityTest(PicardTestCase): @@ -11,3 +14,39 @@ class SimilarityTest(PicardTestCase): self.assertEqual(similarity("K!", "K!"), 1.0) self.assertEqual(similarity("BBB", "AAA"), 0.0) self.assertAlmostEqual(similarity("ABC", "ABB"), 0.7, 1) + + +class Similarity2Test(PicardTestCase): + def test_1(self): + a = b = "a b c" + self.assertEqual(similarity2(a, b), 1.0) + + def test_2(self): + a = "a b c" + b = "A,B•C" + self.assertEqual(similarity2(a, b), 1.0) + + def test_3(self): + a = "a b c" + b = ",A, B •C•" + self.assertEqual(similarity2(a, b), 1.0) + + def test_4(self): + a = "a b c" + b = "c a b" + self.assertEqual(similarity2(a, b), 1.0) + + def test_5(self): + a = "a b c" + b = "a b d" + self.assertAlmostEqual(similarity2(a, b), 0.6, 1) + + def test_6(self): + a = "a b c" + b = "a f d" + self.assertAlmostEqual(similarity2(a, b), 0.3, 1) + + def test_7(self): + a = "abc" + b = "def" + self.assertEqual(similarity2(a, b), 0.0)