fix unit tests

Author: tswast

bigframes/bigquery/_operations/ai.py

@@ -201,7 +201,7 @@ def generate_int(
 
         >>> import bigframes.pandas as bpd
         >>> import bigframes.bigquery as bbq
-                >>> animal = bpd.Series(["Kangaroo", "Rabbit", "Spider"])
+        >>> animal = bpd.Series(["Kangaroo", "Rabbit", "Spider"])
         >>> bbq.ai.generate_int(("How many legs does a ", animal, " have?"))
         0    {'result': 2, 'full_response': '{"candidates":...
         1    {'result': 4, 'full_response': '{"candidates":...
@@ -275,7 +275,7 @@ def generate_double(
 
         >>> import bigframes.pandas as bpd
         >>> import bigframes.bigquery as bbq
-                >>> animal = bpd.Series(["Kangaroo", "Rabbit", "Spider"])
+        >>> animal = bpd.Series(["Kangaroo", "Rabbit", "Spider"])
         >>> bbq.ai.generate_double(("How many legs does a ", animal, " have?"))
         0    {'result': 2.0, 'full_response': '{"candidates...
         1    {'result': 4.0, 'full_response': '{"candidates...
@@ -346,7 +346,7 @@ def if_(
     **Examples:**
         >>> import bigframes.pandas as bpd
         >>> import bigframes.bigquery as bbq
-                >>> us_state = bpd.Series(["Massachusetts", "Illinois", "Hawaii"])
+        >>> us_state = bpd.Series(["Massachusetts", "Illinois", "Hawaii"])
         >>> bbq.ai.if_((us_state, " has a city called Springfield"))
         0     True
         1     True
@@ -395,7 +395,7 @@ def classify(
 
         >>> import bigframes.pandas as bpd
         >>> import bigframes.bigquery as bbq
-                >>> df = bpd.DataFrame({'creature': ['Cat', 'Salmon']})
+        >>> df = bpd.DataFrame({'creature': ['Cat', 'Salmon']})
         >>> df['type'] = bbq.ai.classify(df['creature'], ['Mammal', 'Fish'])
         >>> df
           creature    type
@@ -445,7 +445,7 @@ def score(
 
         >>> import bigframes.pandas as bpd
         >>> import bigframes.bigquery as bbq
-                >>> animal = bpd.Series(["Tiger", "Rabbit", "Blue Whale"])
+        >>> animal = bpd.Series(["Tiger", "Rabbit", "Blue Whale"])
         >>> bbq.ai.score(("Rank the relative weights of ", animal, " on the scale from 1 to 3")) # doctest: +SKIP
         0    2.0
         1    1.0

bigframes/core/tools/datetimes.py

@@ -42,7 +42,7 @@ def to_datetime(
     utc: bool = False,
     format: Optional[str] = None,
     unit: Optional[str] = None,
-    session: Optional[bigframes.session.Session],
+    session: Optional[bigframes.session.Session] = None,
 ) -> Union[pd.Timestamp, datetime, bigframes.series.Series]:
     if isinstance(arg, (int, float, str, datetime, date)):
         return pd.to_datetime(

bigframes/operations/ai.py

@@ -114,7 +114,7 @@ def map(
         **Examples:**
 
             >>> import bigframes.pandas as bpd
-                        >>> bpd.options.experiments.ai_operators = True
+                >>> bpd.options.experiments.ai_operators = True
             >>> bpd.options.compute.ai_ops_confirmation_threshold = 25
 
             >>> import bigframes.ml.llm as llm
@@ -132,7 +132,7 @@ def map(
 
 
             >>> import bigframes.pandas as bpd
-                        >>> bpd.options.experiments.ai_operators = True
+                >>> bpd.options.experiments.ai_operators = True
             >>> bpd.options.compute.ai_ops_confirmation_threshold = 25
 
             >>> import bigframes.ml.llm as llm
@@ -263,7 +263,7 @@ def classify(
         **Examples:**
 
             >>> import bigframes.pandas as bpd
-                        >>> bpd.options.experiments.ai_operators = True
+                >>> bpd.options.experiments.ai_operators = True
             >>> bpd.options.compute.ai_ops_confirmation_threshold = 25
 
             >>> import bigframes.ml.llm as llm
@@ -352,7 +352,7 @@ def join(
         **Examples:**
 
             >>> import bigframes.pandas as bpd
-                        >>> bpd.options.experiments.ai_operators = True
+                >>> bpd.options.experiments.ai_operators = True
             >>> bpd.options.compute.ai_ops_confirmation_threshold = 25
 
             >>> import bigframes.ml.llm as llm
@@ -602,7 +602,7 @@ def sim_join(
         ** Examples: **
 
             >>> import bigframes.pandas as bpd
-                        >>> bpd.options.experiments.ai_operators = True
+                >>> bpd.options.experiments.ai_operators = True
             >>> bpd.options.compute.ai_ops_confirmation_threshold = 25
 
             >>> import bigframes.ml.llm as llm

bigframes/operations/base.py

@@ -135,7 +135,7 @@ def __init__(
         # explicitly chose even if it is None. This is important for the
         # polars backend where the implicit column labels are integers.
         if not isinstance(data, blocks.Block):
-            block = block.with_column_labels([name])
+            block = block.with_column_labels([name or getattr(data, "name", None)])
 
         self._block: blocks.Block = block
 

bigframes/operations/semantics.py

@@ -246,7 +246,7 @@ def cluster_by(
         **Examples:**
 
             >>> import bigframes.pandas as bpd
-                        >>> bpd.options.experiments.semantic_operators = True
+                >>> bpd.options.experiments.semantic_operators = True
             >>> bpd.options.compute.semantic_ops_confirmation_threshold = 25
 
             >>> import bigframes.ml.llm as llm
@@ -319,7 +319,7 @@ def filter(self, instruction: str, model, ground_with_google_search: bool = Fals
         **Examples:**
 
             >>> import bigframes.pandas as bpd
-                        >>> bpd.options.experiments.semantic_operators = True
+                >>> bpd.options.experiments.semantic_operators = True
             >>> bpd.options.compute.semantic_ops_confirmation_threshold = 25
 
             >>> import bigframes.ml.llm as llm
@@ -432,7 +432,7 @@ def map(
         **Examples:**
 
             >>> import bigframes.pandas as bpd
-                        >>> bpd.options.experiments.semantic_operators = True
+                >>> bpd.options.experiments.semantic_operators = True
             >>> bpd.options.compute.semantic_ops_confirmation_threshold = 25
 
             >>> import bigframes.ml.llm as llm
@@ -554,7 +554,7 @@ def join(
         **Examples:**
 
             >>> import bigframes.pandas as bpd
-                        >>> bpd.options.experiments.semantic_operators = True
+                >>> bpd.options.experiments.semantic_operators = True
             >>> bpd.options.compute.semantic_ops_confirmation_threshold = 25
 
             >>> import bigframes.ml.llm as llm
@@ -794,7 +794,7 @@ def top_k(
         **Examples:**
 
             >>> import bigframes.pandas as bpd
-                        >>> bpd.options.experiments.semantic_operators = True
+                >>> bpd.options.experiments.semantic_operators = True
             >>> bpd.options.compute.semantic_ops_confirmation_threshold = 25
 
             >>> import bigframes.ml.llm as llm
@@ -994,7 +994,7 @@ def sim_join(
         ** Examples: **
 
             >>> import bigframes.pandas as bpd
-                        >>> bpd.options.experiments.semantic_operators = True
+                >>> bpd.options.experiments.semantic_operators = True
             >>> bpd.options.compute.semantic_ops_confirmation_threshold = 25
 
             >>> import bigframes.ml.llm as llm

bigframes/pandas/__init__.py

@@ -226,7 +226,6 @@ def to_datetime(
         format=format,
         unit=unit,
     )
-    return bigframes.core.tools.to_datetime()
 
 
 to_datetime.__doc__ = vendored_pandas_datetimes.to_datetime.__doc__

bigframes/session/__init__.py

@@ -2291,6 +2291,12 @@ def read_gbq_object_table(
     # interchangeably.
     # =========================================================================
     def cut(self, *args, **kwargs) -> bigframes.series.Series:
+        """Cuts a BigQuery DataFrames object.
+
+        Included for compatibility between bpd and Session.
+
+        See :func:`bigframes.pandas.cut` for full documentation.
+        """
         import bigframes.core.reshape.tile
 
         return bigframes.core.reshape.tile.cut(
@@ -2299,7 +2305,7 @@ def cut(self, *args, **kwargs) -> bigframes.series.Series:
             **kwargs,
         )
 
-    def DataFrame(self, *args, **kwargs) -> bigframes.dataframe.DataFrame:
+    def DataFrame(self, *args, **kwargs):
         """Constructs a DataFrame.
 
         Included for compatibility between bpd and Session.
@@ -2310,7 +2316,7 @@ def DataFrame(self, *args, **kwargs) -> bigframes.dataframe.DataFrame:
 
         return bigframes.dataframe.DataFrame(*args, session=self, **kwargs)
 
-    def MultiIndex(self, *args, **kwargs) -> bigframes.core.indexes.MultiIndex:
+    def MultiIndex(self, *args, **kwargs):
         """Constructs a MultiIndex.
 
         Included for compatibility between bpd and Session.
@@ -2325,7 +2331,7 @@ def MultiIndex(self, *args, **kwargs) -> bigframes.core.indexes.MultiIndex:
     MultiIndex.from_frame = bigframes.core.indexes.MultiIndex.from_frame  # type: ignore
     MultiIndex.from_arrays = bigframes.core.indexes.MultiIndex.from_arrays  # type: ignore
 
-    def Index(self, *args, **kwargs) -> bigframes.core.indexes.Index:
+    def Index(self, *args, **kwargs):
         """Constructs a Index.
 
         Included for compatibility between bpd and Session.
@@ -2336,7 +2342,7 @@ def Index(self, *args, **kwargs) -> bigframes.core.indexes.Index:
 
         return bigframes.core.indexes.Index(*args, session=self, **kwargs)
 
-    def Series(self, *args, **kwargs) -> bigframes.series.Series:
+    def Series(self, *args, **kwargs):
         """Constructs a Series.
 
         Included for compatibility between bpd and Session.
@@ -2350,6 +2356,12 @@ def Series(self, *args, **kwargs) -> bigframes.series.Series:
     def to_datetime(
         self, *args, **kwargs
     ) -> Union[pandas.Timestamp, datetime.datetime, bigframes.series.Series]:
+        """Converts a BigQuery DataFrames object to datetime dtype.
+
+        Included for compatibility between bpd and Session.
+
+        See :func:`bigframes.pandas.to_datetime` for full documentation.
+        """
         import bigframes.core.tools
 
         return bigframes.core.tools.to_datetime(
@@ -2359,6 +2371,12 @@ def to_datetime(
         )
 
     def to_timedelta(self, *args, **kwargs):
+        """Converts a BigQuery DataFrames object to timedelta/duration dtype.
+
+        Included for compatibility between bpd and Session.
+
+        See :func:`bigframes.pandas.to_timedelta` for full documentation.
+        """
         import bigframes.pandas.core.tools.timedeltas
 
         return bigframes.pandas.core.tools.timedeltas.to_timedelta(

scripts/publish_api_coverage.py

@@ -25,6 +25,7 @@
 import pandas.core.indexes.accessors
 import pandas.core.strings.accessor
 import pandas.core.window.rolling
+import sklearn  # noqa
 
 import bigframes
 import bigframes.core.groupby

tests/unit/conftest.py

@@ -0,0 +1,24 @@
+# Copyright 2025 Google LLC
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+#     http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+
+import pytest
+
+
+@pytest.fixture(scope="session")
+def polars_session():
+    pytest.importorskip("polars")
+
+    from bigframes.testing import polars_session
+
+    return polars_session.TestSession()

tests/unit/test_pandas.py

@@ -64,8 +64,12 @@ def test_method_matches_session(method_name: str):
     pandas_method = getattr(bigframes.pandas, method_name)
     pandas_doc = inspect.getdoc(pandas_method)
     assert pandas_doc is not None, "docstrings are required"
-    assert re.sub(leading_whitespace, "", pandas_doc) == re.sub(
-        leading_whitespace, "", session_doc
+
+    pandas_doc_stripped = re.sub(leading_whitespace, "", pandas_doc)
+    session_doc_stripped = re.sub(leading_whitespace, "", session_doc)
+    assert (
+        pandas_doc_stripped == session_doc_stripped
+        or ":`bigframes.pandas" in session_doc_stripped
     )
 
     # Add `eval_str = True` so that deferred annotations are turned into their
@@ -75,18 +79,20 @@ def test_method_matches_session(method_name: str):
         eval_str=True,
         globals={**vars(bigframes.session), **{"dataframe": bigframes.dataframe}},
     )
-    pandas_signature = inspect.signature(pandas_method, eval_str=True)
-    assert [
-        # Kind includes position, which will be an offset.
-        parameter.replace(kind=inspect.Parameter.POSITIONAL_ONLY)
-        for parameter in pandas_signature.parameters.values()
-    ] == [
+    session_args = [
         # Kind includes position, which will be an offset.
         parameter.replace(kind=inspect.Parameter.POSITIONAL_ONLY)
         for parameter in session_signature.parameters.values()
         # Don't include the first parameter, which is `self: Session`
-    ][
-        1:
+    ][1:]
+    pandas_signature = inspect.signature(pandas_method, eval_str=True)
+    pandas_args = [
+        # Kind includes position, which will be an offset.
+        parameter.replace(kind=inspect.Parameter.POSITIONAL_ONLY)
+        for parameter in pandas_signature.parameters.values()
+    ]
+    assert session_args == pandas_args or ["args", "kwargs"] == [
+        parameter.name for parameter in session_args
     ]
     assert pandas_signature.return_annotation == session_signature.return_annotation