def _setup():
global magic, spark_controller, shell, ipython_display
conf.override_all({})
shell = MagicMock()
ipython_display = MagicMock()
magic = RemoteSparkMagics(shell=None)
magic.shell = shell
magic.ipython_display = ipython_display
spark_controller = MagicMock()
magic.spark_controller = spark_controller
def test_get_livy_kind_covers_all_langs():
for lang in Constants.lang_supported:
RemoteSparkMagics._get_livy_kind(lang)