diff --git a/graphene/test/__init__.py b/graphene/test/__init__.py index 6d46bd57..a613d905 100644 --- a/graphene/test/__init__.py +++ b/graphene/test/__init__.py @@ -18,7 +18,9 @@ def format_execution_result(execution_result, format_error): response = {} if execution_result.errors: - response['errors'] = [format_error(e) for e in execution_result.errors] + response['errors'] = [ + format_error(e) for e in execution_result.errors + ] if not execution_result.invalid: response['data'] = execution_result.data @@ -32,16 +34,14 @@ class Client(object): self.schema = schema self.execute_options = execute_options self.format_error = format_error or default_format_error - + def format_result(self, result): - return format_execution_result( - result, - self.format_error - ) + return format_execution_result(result, self.format_error) def execute(self, *args, **kwargs): - executed = self.schema.execute(*args, **dict(self.execute_options, **kwargs)) + executed = self.schema.execute(*args, + **dict(self.execute_options, **kwargs)) if is_thenable(executed): return Promise.resolve(executed).then(self.format_result) - + return self.format_result(executed)