make sure JSON is callable

tswast · tswast · commit df87a7080379 · 2025-03-10T10:31:38.000-05:00
diff --git a/google/cloud/bigquery/_pyarrow_helpers.py b/google/cloud/bigquery/_pyarrow_helpers.py
@@ -87,7 +87,7 @@ def pyarrow_timestamp():
         "GEOGRAPHY": pyarrow.string,
         "INT64": pyarrow.int64,
         "INTEGER": pyarrow.int64,
-        "JSON": json_arrow_type,
+        "JSON": lambda: json_arrow_type,
         "NUMERIC": pyarrow_numeric,
         "STRING": pyarrow.string,
         "TIME": pyarrow_time,
diff --git a/tests/unit/test__pyarrow_helpers.py b/tests/unit/test__pyarrow_helpers.py
@@ -27,11 +27,11 @@ def module_under_test():
 
 def test_bq_to_arrow_scalars(module_under_test):
     assert (
-        module_under_test.bq_to_arrow_scalars("BIGNUMERIC")
-        == module_under_test.pyarrow_bignumeric
+        module_under_test.bq_to_arrow_scalars("BIGNUMERIC")()
+        == module_under_test.pyarrow_bignumeric()
     )
     assert (
-        module_under_test.bq_to_arrow_scalars("JSON")
+        module_under_test.bq_to_arrow_scalars("JSON")()
         == module_under_test.json_arrow_type
     )
     assert module_under_test.bq_to_arrow_scalars("UNKNOWN_TYPE") is None

Original file line number	Diff line number	Diff line change
`@@ -27,11 +27,11 @@ def module_under_test():`
`27`	`27`
`28`	`28`	`def test_bq_to_arrow_scalars(module_under_test):`
`29`	`29`	`assert (`
`30`		`- module_under_test.bq_to_arrow_scalars("BIGNUMERIC")`
`31`		`- == module_under_test.pyarrow_bignumeric`
	`30`	`+ module_under_test.bq_to_arrow_scalars("BIGNUMERIC")()`
	`31`	`+ == module_under_test.pyarrow_bignumeric()`
`32`	`32`	`)`
`33`	`33`	`assert (`
`34`		`- module_under_test.bq_to_arrow_scalars("JSON")`
	`34`	`+ module_under_test.bq_to_arrow_scalars("JSON")()`
`35`	`35`	`== module_under_test.json_arrow_type`
`36`	`36`	`)`
`37`	`37`	`assert module_under_test.bq_to_arrow_scalars("UNKNOWN_TYPE") is None`