🦉 Updates from OwlBot post-processor

See https://github.com/googleapis/repo-automation-bots/blob/main/packages/owl-bot/README.md

Author: gcf-owl-bot[bot]

bigquery_magics/graph_server.py

@@ -66,7 +66,9 @@ def convert_graph_data(query_results: Dict[str, Dict[str, str]]):
                 if not isinstance(key, str):
                     raise ValueError(f"Expected outer key to be str, got {type(key)}")
                 if not isinstance(value, dict):
-                    raise ValueError(f"Expected outer value to be dict, got {type(value)}")
+                    raise ValueError(
+                        f"Expected outer value to be dict, got {type(value)}"
+                    )
                 column_name = key
                 column_value = value
             else:
@@ -88,7 +90,9 @@ def convert_graph_data(query_results: Dict[str, Dict[str, str]]):
             if not isinstance(value_key, str):
                 raise ValueError(f"Expected inner key to be str, got {type(value_key)}")
             if not isinstance(value_value, str):
-                raise ValueError(f"Expected inner value to be str, got {type(value_value)}")
+                raise ValueError(
+                    f"Expected inner value to be str, got {type(value_value)}"
+                )
             row_index = int(value_key)
             row_json = json.loads(value_value)
 

tests/unit/test_graph_server.py

@@ -222,22 +222,20 @@ def test_convert_nongraph_json():
 )
 def test_convert_outer_key_not_string():
     result = convert_graph_data(
-            {
-                0: {
-                    '0': json.dumps({"foo": 1, "bar": 2}),
-                }
-            })
+        {
+            0: {
+                "0": json.dumps({"foo": 1, "bar": 2}),
+            }
+        }
+    )
     assert result == {"error": "Expected outer key to be str, got <class 'int'>"}
 
 
 @pytest.mark.skipif(
     graph_visualization is None, reason="Requires `spanner-graph-notebook`"
 )
 def test_convert_outer_value_not_dict():
-    result = convert_graph_data(
-            {
-                'result': 0
-            })
+    result = convert_graph_data({"result": 0})
     assert result == {"error": "Expected outer value to be dict, got <class 'int'>"}
 
 
@@ -246,11 +244,12 @@ def test_convert_outer_value_not_dict():
 )
 def test_convert_inner_key_not_string():
     result = convert_graph_data(
-            {
-                'result': {
-                    0: json.dumps({"foo": 1, "bar": 2}),
-                }
-            })
+        {
+            "result": {
+                0: json.dumps({"foo": 1, "bar": 2}),
+            }
+        }
+    )
     assert result == {"error": "Expected inner key to be str, got <class 'int'>"}
 
 
@@ -259,11 +258,12 @@ def test_convert_inner_key_not_string():
 )
 def test_convert_inner_value_not_string():
     result = convert_graph_data(
-            {
-                'result': {
-                    '0': 1,
-                }
-            })
+        {
+            "result": {
+                "0": 1,
+            }
+        }
+    )
     assert result == {"error": "Expected inner value to be str, got <class 'int'>"}
 
 
@@ -278,19 +278,22 @@ def test_convert_one_column_one_row_two_columns():
             },
             "result2": {
                 "0": json.dumps(row_alex_owns_account),
-            }
-
+            },
         }
     )
-    assert result == {"error": "Query has multiple columns - graph visualization not supported"}
+    assert result == {
+        "error": "Query has multiple columns - graph visualization not supported"
+    }
 
 
 @pytest.mark.skipif(
     graph_visualization is None, reason="Requires `spanner-graph-notebook`"
 )
 def test_convert_empty_dict():
     result = convert_graph_data({})
-    assert result == {"error": "query result with no columns is not supported for graph visualization"}
+    assert result == {
+        "error": "query result with no columns is not supported for graph visualization"
+    }
 
 
 class TestGraphServer(unittest.TestCase):