fix prerelease tests

Author: tswast

packages/bigframes/noxfile.py

@@ -605,11 +605,11 @@ def prerelease(session: nox.sessions.Session, tests_path, extra_pytest_options=(
         # Workaround https://github.com/googleapis/python-db-dtypes-pandas/issues/178
         "db-dtypes",
         # Ensure we catch breaking changes in the client libraries early.
-        "git+https://github.com/googleapis/python-bigquery.git#egg=google-cloud-bigquery",
+        "git+https://github.com/googleapis/google-cloud-python.git#egg=google-cloud-bigquery&subdirectory=packages/google-cloud-bigquery",
         "--upgrade",
         "-e",
         "git+https://github.com/googleapis/google-cloud-python.git#egg=google-cloud-bigquery-storage&subdirectory=packages/google-cloud-bigquery-storage",
-        "git+https://github.com/googleapis/python-bigquery-pandas.git#egg=pandas-gbq",
+        "git+https://github.com/googleapis/google-cloud-python.git#egg=pandas-gbq&subdirectory=packages/pandas-gbq",
     )
 
     # Print out prerelease package versions.

packages/bigframes/tests/system/small/test_series.py

@@ -33,6 +33,7 @@
 import bigframes.pandas
 import bigframes.series as series
 import bigframes.testing
+import bigframes.testing.utils
 from bigframes.testing.utils import (
     assert_frame_equal,
     assert_series_equal,
@@ -1232,23 +1233,12 @@ def test_divmods_series(scalars_dfs, col_x, col_y, method):
         scalars_pandas_df[col_y]
     )
     # BigQuery's mod functions return NUMERIC values for non-INT64 inputs.
-    if bf_div_result.dtype == pd.Int64Dtype():
-        bigframes.testing.utils.assert_series_equal(
-            pd_div_result, bf_div_result.to_pandas(), check_dtype=False
-        )
-    else:
-        bigframes.testing.utils.assert_series_equal(
-            pd_div_result, bf_div_result.astype("Float64").to_pandas()
-        )
-
-    if bf_mod_result.dtype == pd.Int64Dtype():
-        bigframes.testing.utils.assert_series_equal(
-            pd_mod_result, bf_mod_result.to_pandas()
-        )
-    else:
-        bigframes.testing.utils.assert_series_equal(
-            pd_mod_result, bf_mod_result.astype("Float64").to_pandas()
-        )
+    bigframes.testing.utils.assert_series_equal(
+        pd_div_result, bf_div_result.to_pandas(), check_dtype=False
+    )
+    bigframes.testing.utils.assert_series_equal(
+        pd_mod_result, bf_mod_result.to_pandas(), check_dtype=False
+    )
 
 
 @pytest.mark.parametrize(