Skip to content

Commit 3a88427

Browse files
committed
temporarily avoid pyarrow.json_
1 parent 8c57082 commit 3a88427

File tree

1 file changed

+11
-9
lines changed

1 file changed

+11
-9
lines changed

tests/system/test_to_gbq.py

Lines changed: 11 additions & 9 deletions
Original file line numberDiff line numberDiff line change
@@ -588,10 +588,11 @@ def test_series_round_trip(
588588
'"string"',
589589
],
590590
dtype=(
591-
pandas.ArrowDtype(pyarrow.json_(pyarrow.string()))
592-
if hasattr(pandas, "ArrowDtype")
593-
and hasattr(pyarrow, "json_")
594-
else "object"
591+
pandas.ArrowDtype(db_dtypes.JSONArrowType())
592+
# pandas.ArrowDtype(pyarrow.json_(pyarrow.string()))
593+
# if hasattr(pandas, "ArrowDtype")
594+
# and hasattr(pyarrow, "json_")
595+
# else "object"
595596
),
596597
),
597598
},
@@ -603,7 +604,7 @@ def test_series_round_trip(
603604
"row_num": [0, 1, 2, 3, 4],
604605
"json": pandas.Series(
605606
[
606-
'{"key": "value"}',
607+
'{"key":"value"}',
607608
None,
608609
"123",
609610
"[123]",
@@ -617,10 +618,11 @@ def test_series_round_trip(
617618
# though, we should probably move the BQ -> pandas type mapping to this
618619
# package so the logic can be consolidated soon.
619620
dtype=(
620-
pandas.ArrowDtype(pyarrow.json_(pyarrow.string()))
621-
if hasattr(pandas, "ArrowDtype")
622-
and hasattr(pyarrow, "json_")
623-
else "object"
621+
pandas.ArrowDtype(db_dtypes.JSONArrowType())
622+
# pandas.ArrowDtype(pyarrow.json_(pyarrow.string()))
623+
# if hasattr(pandas, "ArrowDtype")
624+
# and hasattr(pyarrow, "json_")
625+
# else "object"
624626
),
625627
),
626628
},

0 commit comments

Comments
 (0)