def setUp(self):
mongoConn = MongoConnection()
self.storage = Storage(mongoConn)
reader = TextReader()
self.calculator = TagStudy(self.storage, reader)
class TagStudyTests(unittest.TestCase):
def setUp(self):
mongoConn = MongoConnection()
self.storage = Storage(mongoConn)
reader = TextReader()
self.calculator = TagStudy(self.storage, reader)
def tearDown(self):
self.calculator = None
def test_learn_new_tag(self):
tags = ['tag1']
text = 'Hello World!'
self.calculator.learn(tags, text)
self.check_tag_words('tag1', {'hello': 1, 'world': 1})
def test_learn_few_tags(self):
tags = ['tag1', 'tag2']
text = 'Hello World!'
self.calculator.learn(tags, text)
self.check_tag_words('tag1', {'hello': 1, 'world': 1})
self.check_tag_words('tag2', {'hello': 1, 'world': 1})
def test_learn_merge_tags(self):
tags = ['tag1']
text = 'Hello World!'
self.calculator.learn(tags, text)
self.check_tag_words('tag1', {'hello': 1, 'world': 1}, remove=False)
text = 'Hello Ruslan!'
self.calculator.learn(tags, text)
self.check_tag_words('tag1', {'hello': 2, 'world': 1, 'ruslan': 1})
def test_learn_tags_words_parts(self):
tags = ['tag1']
text = 'Hello World! World is nice, so it is a pleasure to say hello to the world'
self.calculator.learn(tags, text)
words = self.storage.get_tag_words('tag1')
try:
foundHello = False
foundWorld = False
foundNice = False
for word in words:
if word.word == 'hello':
self.assertEqual(2, word.count)
self.assertEqual(float(2.0 / 8), word.part)
foundHello = True
if word.word == 'world':
self.assertEqual(3, word.count)
self.assertEqual(3.0 / 8, word.part)
foundWorld = True
if word.word == 'nice':
self.assertEqual(1, word.count)
self.assertEqual(1.0 / 8, word.part)
foundNice = True
self.assertTrue(foundHello, 'Hello is not found')
self.assertTrue(foundWorld, 'World is not found')
self.assertTrue(foundNice, 'Nice is not found')
finally:
self.storage.remove_tag('tag1')
def check_tag_words(self, tag, checkWords, remove=True):
words = self.storage.get_tag_words(tag)
for key, value in checkWords.items():
self.assertWordInWithCount(words, key, value)
if remove: self.storage.remove_tag(tag)
def assertWordInWithCount(self, words, word, count):
for each in words:
if each.word == word:
self.assertEqual(count, each.count, "Wrong count")
return
self.fail("Word is not found")
# -----------------------------------------------------
def test_forget_new_tag(self):
try:
tags = ['tag1']
text = 'Hello World!'
self.calculator.learn(tags, text)
self.check_tag_words('tag1', {'hello': 1, 'world': 1}, remove=False)
self.calculator.forget(tags, text)
words = self.storage.get_tag_words('tag1')
self.assertEqual(0, len(words))
finally:
self.storage.remove_tag("tag1")
def test_forget_merge_tags(self):
tags = ['tag1']
text = 'Hello World!'
self.calculator.learn(tags, text)
self.check_tag_words('tag1', {'hello': 1, 'world': 1}, remove=False)
text = 'Hello Ruslan!'
self.calculator.learn(tags, text)
self.check_tag_words('tag1', {'hello': 2, 'world': 1, 'ruslan': 1}, remove=False)
self.calculator.forget(tags, text)
self.check_tag_words('tag1', {'hello': 1, 'world': 1})
