|
2 | 2 |
|
3 | 3 | from strsimpy.overlap_coefficient import OverlapCoefficient |
4 | 4 |
|
| 5 | + |
5 | 6 | class TestOverlapCoefficient(unittest.TestCase): |
6 | 7 |
|
7 | 8 | def test_overlap_coefficient_onestringissubsetofother_return0(self): |
8 | 9 | sim = OverlapCoefficient(3) |
9 | | - s1,s2 = "eat","eating" |
10 | | - actual = sim.distance(s1,s2) |
11 | | - print("distance: {:.4}\t between '{}' and '{}'".format(str(actual), s1,s2)) |
12 | | - self.assertEqual(0,actual) |
| 10 | + s1, s2 = "eat", "eating" |
| 11 | + actual = sim.distance(s1, s2) |
| 12 | + print("distance: {:.4}\t between '{}' and '{}'".format(str(actual), s1, s2)) |
| 13 | + self.assertEqual(0, actual) |
13 | 14 |
|
14 | 15 | def test_overlap_coefficient_onestringissubset_return1(self): |
15 | 16 | sim = OverlapCoefficient(3) |
16 | | - s1,s2 = "eat","eating" |
17 | | - actual = sim.similarity(s1,s2) |
18 | | - print("strsim: {:.4}\t between '{}' and '{}'".format(str(actual), s1,s2)) |
19 | | - self.assertEqual(1,actual) |
| 17 | + s1, s2 = "eat", "eating" |
| 18 | + actual = sim.similarity(s1, s2) |
| 19 | + print("strsim: {:.4}\t between '{}' and '{}'".format(str(actual), s1, s2)) |
| 20 | + self.assertEqual(1, actual) |
20 | 21 |
|
21 | 22 | def test_overlap_coefficient_onestringissubsetofother_return1(self): |
22 | 23 | sim = OverlapCoefficient(3) |
23 | | - s1,s2 = "eat","eating" |
24 | | - actual = sim.similarity(s1,s2) |
25 | | - print("strsim: {:.4}\t between '{}' and '{}'".format(str(actual), s1,s2)) |
26 | | - self.assertEqual(1,actual) |
| 24 | + s1, s2 = "eat", "eating" |
| 25 | + actual = sim.similarity(s1, s2) |
| 26 | + print("strsim: {:.4}\t between '{}' and '{}'".format(str(actual), s1, s2)) |
| 27 | + self.assertEqual(1, actual) |
27 | 28 |
|
28 | 29 | def test_overlap_coefficient_halfsimilar_return1(self): |
29 | 30 | sim = OverlapCoefficient(2) |
30 | | - s1,s2 = "car","bar" |
31 | | - self.assertEqual(1/2,sim.similarity(s1,s2)) |
32 | | - self.assertEqual(1/2,sim.distance(s1,s2)) |
| 31 | + s1, s2 = "car", "bar" |
| 32 | + self.assertEqual(1 / 2, sim.similarity(s1, s2)) |
| 33 | + self.assertEqual(1 / 2, sim.distance(s1, s2)) |
| 34 | + |
33 | 35 |
|
34 | 36 | if __name__ == "__main__": |
35 | 37 | unittest.main() |
0 commit comments