diff --git a/test_utils.py b/test_utils.py new file mode 100644 index 000000000..66b0facdf --- /dev/null +++ b/test_utils.py @@ -0,0 +1,15 @@ +import unittest +from utils import replace_most_similar_chunk + +class TestUtils(unittest.TestCase): + def test_replace_most_similar_chunk(self): + whole = "This is a sample text.\nAnother line of text.\nYet another line." + part = "sample text" + replace = "replaced text" + expected_output = "This is a replaced text.\nAnother line of text.\nYet another line." + + result = replace_most_similar_chunk(whole, part, replace) + self.assertEqual(result, expected_output) + +if __name__ == "__main__": + unittest.main()