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)