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)