def test_word_storage_get_id_not_in_storage(self):
"""
word not in storage
"""
word_storage = WordStorage()
word_storage.storage = {'word': 1}
self.assertRaises(KeyError, word_storage.get_id, 'word2')
def test_word_storage_get_id_not_str(self):
"""
id is not str get_id
"""
word_storage = WordStorage()
word_storage.storage = {'word': 1}
self.assertRaises(ValueError, word_storage.get_id, 123)
def test_word_storage_get_id_none(self):
"""
get_id none
"""
word_storage = WordStorage()
word_storage.storage = {'word': 1}
self.assertRaises(ValueError, word_storage.get_id, None)
def test_word_storage_get_id_ideal(self):
"""
ideal case for get_id
"""
word_storage = WordStorage()
word_storage.storage = {'word': 1}
expected = 1
actual = word_storage.get_id('word')
self.assertEqual(expected, actual)
def test_word_storage_get_word_incorrect_num(self): # new test
"""
id is not incorrect num
"""
word_storage = WordStorage()
word_storage.storage = {'word': 1}
bad_inputs = (2.8, -3, 0)
for bad_input in bad_inputs:
self.assertRaises(ValueError, word_storage.get_word, bad_input)
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)
