import unittest

from strsimpy.overlap_coefficient import OverlapCoefficient

class TestOverlapCoefficient(unittest.TestCase):

    def test_overlap_coefficient_onestringissubsetofother_return0(self):
        sim = OverlapCoefficient(3)
        s1,s2 = "eat","eating"
        actual = sim.distance(s1,s2)
        print("distance: {:.4}\t between '{}' and '{}'".format(str(actual), s1,s2))
        self.assertEqual(0,actual)

    def test_overlap_coefficient_onestringissubset_return1(self):
        sim = OverlapCoefficient(3)
        s1,s2 = "eat","eating"
        actual = sim.similarity(s1,s2)
        print("strsim: {:.4}\t between '{}' and '{}'".format(str(actual), s1,s2))
        self.assertEqual(1,actual)

    def test_overlap_coefficient_onestringissubsetofother_return1(self):
        sim = OverlapCoefficient(3)
        s1,s2 = "eat","eating"
        actual = sim.similarity(s1,s2)
        print("strsim: {:.4}\t between '{}' and '{}'".format(str(actual), s1,s2))
        self.assertEqual(1,actual)

    def test_overlap_coefficient_halfsimilar_return1(self):
        sim = OverlapCoefficient(2)
        s1,s2 = "car","bar"
        self.assertEqual(1/2,sim.similarity(s1,s2))
        self.assertEqual(1/2,sim.distance(s1,s2))

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