def test_word_storage_put_word_ideal(self):
"""
word is added to storage
"""
word_storage = WordStorage()
word = 'word'
actual = word_storage._put_word(word)
self.assertTrue(word in word_storage.storage)
expected = word_storage.get_id(word)
self.assertEqual(expected, actual)
def test_word_storage_put_word_existing(self):
"""
existing word is not added to storage
"""
word_storage = WordStorage()
word = 'word'
word_storage.storage = {'word': 1}
expected = 1
actual = word_storage._put_word(word)
self.assertEqual(word_storage.storage, {'word': 1})
self.assertEqual(expected, actual)
