def test_get_flavor_import_error(self):
import builtins
_original_import = builtins.__import__
def error_raiser(*args, **kwargs):
raise ImportError('fake import error')
builtins.__import__ = error_raiser
flavor = get_flavor()
builtins.__import__ = _original_import
self.assertEqual(flavor, "jovian")
def test_get_flavor(self):
self.assertEqual(get_flavor(), "jovian")
def test_get_flavor():
assert get_flavor() == "jovian"