test_separate_operations: refactor tests to look like snapshots

Replicates graphql/graphql-js@93e26db

Author: Cito

tests/utilities/test_separate_operations.py

@@ -46,12 +46,15 @@ def separates_one_ast_into_multiple_maintaining_document_order():
             """
         )
 
-        separated_asts = separate_operations(ast)
+        separated_asts = {
+            name: print_ast(node) for name, node in separate_operations(ast).items()
+        }
 
         assert list(separated_asts) == ["", "One", "Two"]
 
-        assert print_ast(separated_asts[""]) == dedent(
-            """
+        assert separated_asts == {
+            "": dedent(
+                """
             {
               ...Y
               ...X
@@ -65,10 +68,9 @@ def separates_one_ast_into_multiple_maintaining_document_order():
               fieldY
             }
             """
-        )
-
-        assert print_ast(separated_asts["One"]) == dedent(
-            """
+            ),
+            "One": dedent(
+                """
             query One {
               foo
               bar
@@ -89,10 +91,9 @@ def separates_one_ast_into_multiple_maintaining_document_order():
               something
             }
             """
-        )
-
-        assert print_ast(separated_asts["Two"]) == dedent(
-            """
+            ),
+            "Two": dedent(
+                """
             fragment A on T {
               field
               ...B
@@ -112,7 +113,8 @@ def separates_one_ast_into_multiple_maintaining_document_order():
               something
             }
             """
-        )
+            ),
+        }
 
     def survives_circular_dependencies():
         ast = parse(
@@ -135,12 +137,15 @@ def survives_circular_dependencies():
             """
         )
 
-        separated_asts = separate_operations(ast)
+        separated_asts = {
+            name: print_ast(node) for name, node in separate_operations(ast).items()
+        }
 
         assert list(separated_asts) == ["One", "Two"]
 
-        assert print_ast(separated_asts["One"]) == dedent(
-            """
+        assert separated_asts == {
+            "One": dedent(
+                """
             query One {
               ...A
             }
@@ -153,10 +158,9 @@ def survives_circular_dependencies():
               ...A
             }
             """
-        )
-
-        assert print_ast(separated_asts["Two"]) == dedent(
-            """
+            ),
+            "Two": dedent(
+                """
             fragment A on T {
               ...B
             }
@@ -169,4 +173,5 @@ def survives_circular_dependencies():
               ...B
             }
             """
-        )
+            ),
+        }