From 45d1d2eb56ba54f2cfe2161aefac9fc6d76c9692 Mon Sep 17 00:00:00 2001
From: Jonathan Kim <jkimbo@gmail.com>
Date: Thu, 26 Dec 2019 11:49:29 +0000
Subject: [PATCH] Fix tests

---
 graphene/relay/tests/test_node.py        | 21 ++++++++++++---------
 graphene/relay/tests/test_node_custom.py |  6 ++++--
 2 files changed, 16 insertions(+), 11 deletions(-)

diff --git a/graphene/relay/tests/test_node.py b/graphene/relay/tests/test_node.py
index c322b1a3..62fd31a3 100644
--- a/graphene/relay/tests/test_node.py
+++ b/graphene/relay/tests/test_node.py
@@ -183,15 +183,18 @@ def test_str_schema():
 
         type RootQuery {
           first: String
-
-          """The ID of the object"""
-          node(id: ID!): Node
-
-          """The ID of the object"""
-          onlyNode(id: ID!): MyNode
-
-          """The ID of the object"""
-          onlyNodeLazy(id: ID!): MyNode
+          node(
+            """The ID of the object"""
+            id: ID!
+          ): Node
+          onlyNode(
+            """The ID of the object"""
+            id: ID!
+          ): MyNode
+          onlyNodeLazy(
+            """The ID of the object"""
+            id: ID!
+          ): MyNode
         }
         '''
     )
diff --git a/graphene/relay/tests/test_node_custom.py b/graphene/relay/tests/test_node_custom.py
index 773be48f..6f28eb66 100644
--- a/graphene/relay/tests/test_node_custom.py
+++ b/graphene/relay/tests/test_node_custom.py
@@ -78,8 +78,10 @@ def test_str_schema_correct():
         }
 
         type RootQuery {
-          """The ID of the object"""
-          node(id: ID!): Node
+          node(
+            """The ID of the object"""
+            id: ID!
+          ): Node
         }
 
         type User implements Node {