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")