def test_error_for_remove_vowels(self):
"""Test that a typer error occurs when a string is not passed"""
with self.assertRaises(TypeError):
main.remove_vowels(5)
main.remove_vowels(['hello', 'world'])
main.remove_vowels({'hi': 'whats up', 'num' : 90})
def test_vowels_removed(self):
"""Test that the vowels are actually removed from the string"""
self.assertEqual(main.remove_vowels('this is test'),
['t', 'h', 's', ' ', 's', ' ', 't', 's', 't'])
