test: Remove _get_nodes

Author: abawchen

graphene_mongo/tests/test_relay_query.py

@@ -12,10 +12,6 @@
 from ..fields import MongoengineConnectionField
 
 
-def _get_nodes(data, key):
-    return map(lambda edge: edge['node'], data[key]['edges'])
-
-
 def test_should_query_reporter(fixtures):
 
     class Query(graphene.ObjectType):
@@ -555,13 +551,13 @@ class Query(graphene.ObjectType):
         'editors': {
             'edges': [
                 {
-                    'cursor': 'xxx',
+                    'cursor': 'YXJyYXljb25uZWN0aW9uOjA=',
                     'node': {
                         'firstName': 'Penny'
                     }
                 },
                 {
-                    'cursor': 'xxx',
+                    'cursor': 'YXJyYXljb25uZWN0aW9uOjE=',
                     'node': {
                         'firstName': 'Grant'
                     }
@@ -570,17 +566,16 @@ class Query(graphene.ObjectType):
             'pageInfo': {
                 'hasNextPage': True,
                 'hasPreviousPage': False,
-                'startCursor': 'xxx',
-                'endCursor': 'xxx'
+                'startCursor': 'YXJyYXljb25uZWN0aW9uOjA=',
+                'endCursor': 'YXJyYXljb25uZWN0aW9uOjE='
             }
         }
     }
     schema = graphene.Schema(query=Query)
     result = schema.execute(query)
 
     assert not result.errors
-    assert all(item in _get_nodes(result.data, 'editors')
-               for item in _get_nodes(expected, 'editors'))
+    assert result.data == expected
 
 
 def test_should_after(fixtures):