From fcc59bcb64cb69f2ba4906773aa0e55c15f8dc62 Mon Sep 17 00:00:00 2001 From: dongfangtianyu Date: Sat, 6 Apr 2019 22:17:42 +0800 Subject: [PATCH] reformat code --- tests/schemas/test_openapi.py | 10 +++++----- tests/schemas/views.py | 5 ++--- 2 files changed, 7 insertions(+), 8 deletions(-) diff --git a/tests/schemas/test_openapi.py b/tests/schemas/test_openapi.py index 529c0756a..4c64810f7 100644 --- a/tests/schemas/test_openapi.py +++ b/tests/schemas/test_openapi.py @@ -168,18 +168,18 @@ class TestGenerator(TestCase): assert 'paths' in schema def test_serializer_datefield(self): - patterns = [ - url(r'^example/?$', views.ExampleGenericViewSet.as_view({"get":"get"})), + url(r'^example/?$', views.ExampleGenericViewSet.as_view({"get": "get"})), ] generator = SchemaGenerator(patterns=patterns) request = create_request('/') schema = generator.get_schema(request=request) - response_schema = schema['paths']['/example/']['get']['responses']['200']['content']['application/json']['schema']['properties'] + response = schema['paths']['/example/']['get']['responses'] + response_schema = response['200']['content']['application/json']['schema']['properties'] - assert response_schema['date']['type'] == response_schema['datetime']['type']== 'string' + assert response_schema['date']['type'] == response_schema['datetime']['type'] == 'string' assert response_schema['date']['format'] == 'date' - assert response_schema['datetime']['format'] == 'date-time' \ No newline at end of file + assert response_schema['datetime']['format'] == 'date-time' diff --git a/tests/schemas/views.py b/tests/schemas/views.py index 5a7bae305..c2c8b7856 100644 --- a/tests/schemas/views.py +++ b/tests/schemas/views.py @@ -23,7 +23,6 @@ class ExampleDetailView(APIView): class ExampleGenericViewSet(GenericViewSet): - class ExampleSerializer(serializers.Serializer): date = serializers.DateField() datetime = serializers.DateTimeField() @@ -34,5 +33,5 @@ class ExampleGenericViewSet(GenericViewSet): from datetime import datetime now = datetime.now() - serializer = self.get_serializer(data=now.date(),datetime=now) - return Response(serializer.data) \ No newline at end of file + serializer = self.get_serializer(data=now.date(), datetime=now) + return Response(serializer.data)