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)