def testTypos(self):
keyword_search = KeywordSearch()
keyword_search.add(['dravid', 'david', 'age'], 0)
keyword_search.add(['sachin', 'tendulkar'], 1)
self.assertEqual(keyword_search.correctTypo('drvid'),
{'dravid', 'david'})
self.assertEqual(keyword_search.correctTypo('tendkar'), {'tendulkar'})
self.assertEqual(keyword_search.correctTypo('schin'), {'sachin'})
def testAddingKeywords(self):
keyword_search = KeywordSearch()
keyword_search.add(['dravid', 'age'], 0)
keyword_search.add(['dravid', 'batting_average'], 1)
keyword_search.add(['sachin', 'batting_average'], 2)
out = keyword_search.search(['dravid'])
self.assertEqual(out, [0, 1])
out = keyword_search.search(['batting_average', 'dravid'])
self.assertEqual(out, [1])
out = keyword_search.search(['What', 'is', 'dravid', 'age'])
self.assertEqual(out, [0])
out = keyword_search.search(['sachin', 'dravid'])
self.assertEqual(out, [0, 1, 2])
def testRemovingKeywords(self):
keyword_search = KeywordSearch()
keyword_search.add(['hello'], 0)
keyword_search.add(['world'], 1)
out = keyword_search.search(['hello'])
self.assertEqual(out, [0])
keyword_search.discard(['hello', 'world'], 0)
out = keyword_search.search(['hello'])
self.assertEqual(out, [])
out = keyword_search.search(['world'])
self.assertEqual(out, [1])
def testMisspelledKeywords(self):
keyword_search = KeywordSearch()
keyword_search.add(['dravid', 'age'], 10)
keyword_search.add(['dravid', 'batting', 'average'], 11)
out = keyword_search.search('What is dravid agge'.split())
self.assertEqual(out, [10])
out = keyword_search.search('What is battting avelage'.split())
self.assertEqual(out, [11])
def testEmptyKeywordSearch(self):
keyword_search = KeywordSearch()
out = keyword_search.search(['hello'])
self.assertEqual(len(out), 0)
def testTyposSmallWords(self):
keyword_search = KeywordSearch()
keyword_search.add(['load', 'id'], 0)
self.assertEqual(keyword_search.correctTypo('good'), set())
self.assertEqual(keyword_search.correctTypo('is'), set())
self.assertEqual(keyword_search.correctTypo('ld'), set())
self.assertEqual(keyword_search.correctTypo('loaad'), {'load'})
self.assertEqual(keyword_search.correctTypo('lod'), {'load'})
keyword_search.add(['tendulkar', 'fine'], 0)
self.assertEqual(keyword_search.correctTypo('fines'), {'fine'})
self.assertEqual(keyword_search.correctTypo('finnes'), {'fine'})
self.assertEqual(keyword_search.correctTypo('tendkar'), {'tendulkar'})
self.assertEqual(keyword_search.correctTypo('tendakar'), set())
self.assertEqual(keyword_search.correctTypo('line'), {'fine'})
self.assertEqual(keyword_search.correctTypo('cone'), set())
def testKnownWords(self):
keyword_search = KeywordSearch()
keyword_search.add(['dravid', 'age'], 0)
keyword_search.add(['sachin', 'tendulkar'], 1)
out_set = keyword_search.known({'sachin', 'age', 'hello', 'dravid'})
self.assertEqual(out_set, {'sachin', 'dravid', 'age'})