def test_is_final_status(self):
kind = "scala"
http_client = MagicMock()
_t_config_hook({
"status_sleep_seconds": 0.01,
"statement_sleep_seconds": 0.01
})
session = LivySession(http_client, kind, "-1", False)
_t_config_hook({})
assert not session.is_final_status("idle")
assert not session.is_final_status("starting")
assert not session.is_final_status("busy")
assert session.is_final_status("dead")
assert session.is_final_status("error")
