def test_serialize(self): url = "url" username = "******" password = "******" connection_string = get_connection_string(url, username, password) http_client = MagicMock() http_client.connection_string = connection_string kind = "scala" _t_config_hook({ "status_sleep_seconds": 0.01, "statement_sleep_seconds": 0.01 }) session = LivySession(http_client, kind, "-1", False) _t_config_hook({}) serialized = session.get_state().to_dict() assert serialized["connectionstring"] == connection_string assert serialized["id"] == "-1" assert serialized["language"] == kind assert serialized["sqlcontext"] == False assert serialized["version"] == "0.0.0" assert len(serialized.keys()) == 5