def testReadExpansionFileWithSuffixes(self):
"""Tests TextPreprocess reads csv files correctly and adds suffixes."""
processor = TextPreprocess()
suffixes = ["", "s", "'s"]
abbreviations = processor.readExpansionFile("abbreviations.csv", suffixes)
expectedAbbreviations = {"wfh": "work from home",
"wfhs": "work from homes",
"wfh's": "work from home's"}
self.assertEqual(abbreviations, expectedAbbreviations)
def testReadExpansionFileNoSuffixes(self):
"""Tests TextPreprocess reads csv files correctly."""
processor = TextPreprocess()
abbreviations = processor.readExpansionFile("abbreviations.csv")
expectedAbbreviations = {"wfh": "work from home"}
self.assertEqual(abbreviations, expectedAbbreviations)
