class DictonaryTests(unittest.TestCase):
def setUp(self):
self.testDic = Dictionary('../words')
self.expectedTestFriends = ('lest','vest','telt','tests','teste','text','nest','teat','rest','testa','testy','fest','pest','tost','jest','gest','yest','hest','tent')
self.expectedLeviathens = set()
def test_size(self):
self.assertEqual(len(self.testDic._dictonary), 380645, "The size of the dictionary is off. Check file and constructor")
def test_levenshtein(self):
self.assertEqual(len(self.testDic._levenshtein('test')), 238)
#self.assertItemsEqual(self.testDic._leviathens('test').sort(), self.expectedLeviathens,"Set of leviathens don't match")
def test_friends(self):
self.assertEqual(len(self.testDic.friends('test')), 19)
self.assertItemsEqual(self.testDic.friends('test'), self.expectedTestFriends )
def test_networkSize(self):
self.assertEqual(len(self.testDic.network('test')), 64413)
pass
'''
Created on Jan 14, 2012
@author: matt
'''
from time import time
from Dictionary import Dictionary
testDic = Dictionary("../words")
testDic.VERBOSE = True
testlistDic = testDic._dictonary
start = time()
levns = testDic._levenshtein("test")
#for levn in levns:
# print levn
end = time()
print 'found ' + str(len(levns)) + ' leviathens in ' + str(end - start) + "for the word 'test'"
start = time()
friends = testDic.friends('test')
#for friend in friends:
# print friend
end = time()
print 'found ' + str(len(friends)) + ' friends in ' + str(end-start) + "for the word 'test'"
start = time()
network = testDic.network('test')
end = time()
