def testUnigrams(self):
analysis = Analysis("Hello, a world!")
result = quantify_variant(analysis, 0)
expected = {
"a": 1 / 15.0,
"h": 1 / 15.0,
"e": 1 / 15.0,
"l": 3 / 15.0,
"o": 2 / 15.0,
"w": 1 / 15.0,
"r": 1 / 15.0,
"d": 1 / 15.0,
}
self.assertSparseDictEqual(expected, result)
def testBigrams(self):
analysis = Analysis("Hello, a world!")
result = quantify_variant(analysis, 1)
expected = {
"<h": 1 / 15.0,
"he": 1 / 15.0,
"el": 1 / 15.0,
"ll": 1 / 15.0,
"lo": 1 / 15.0,
"o>": 1 / 15.0,
"<a": 1 / 15.0,
"a>": 1 / 15.0,
"<w": 1 / 15.0,
"wo": 1 / 15.0,
"or": 1 / 15.0,
"rl": 1 / 15.0,
"ld": 1 / 15.0,
"d>": 1 / 15.0,
}
self.assertSparseDictEqual(expected, result)