Skip to content

Add variance checks to aggregation tests and type assertions #1546

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Open
roman-y-wu wants to merge 7 commits into
base: main
Choose a base branch
from
+58 −6
Open

Add variance checks to aggregation tests and type assertions #1546

Show file tree
Hide file tree
Changes from all commits
Commits
Show all changes
7 commits
Select commit Hold shift + click to select a range
42e67a5
Add variance checks to aggregation tests and type assertions
roman-y-wu Dec 13, 2025
adb98a5
Undo the tests/_typing.py change
roman-y-wu Dec 13, 2025
48b7070
TYP: Fix Series.var() return type annotations:
roman-y-wu Dec 16, 2025
5f8c5b2
GH1546 PR Feedback
loicdiridollou Dec 19, 2025
4985c34
Merge branch 'main' of github.com:pandas-dev/pandas-stubs
loicdiridollou Dec 19, 2025
60eaed7
Update tests/series/test_agg.py
loicdiridollou Dec 19, 2025
854b965
Fix type checking for series variance function
loicdiridollou Dec 19, 2025
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
32 changes: 30 additions & 2 deletions pandas-stubs/core/series.pyi
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -4636,14 +4636,42 @@ class Series(IndexOpsMixin[S1], ElementOpsMixin[S1], NDFrame):
**kwargs: Any,
) -> np_1darray: ...
def tolist(self) -> list[S1]: ...
@overload
def var(
self,
self: Series[Never],
axis: AxisIndex | None = 0,
skipna: _bool | None = True,
ddof: int = 1,
numeric_only: _bool = False,
**kwargs: Any,
) -> Scalar: ...
) -> float: ...
@overload
def var(
self: Series[Timedelta] | Series[Timestamp],
axis: AxisIndex | None = 0,
skipna: _bool | None = True,
ddof: int = 1,
numeric_only: _bool = False,
**kwargs: Any,
) -> Never: ...
@overload
def var(
self: Series[complex],
axis: AxisIndex | None = 0,
skipna: _bool | None = True,
ddof: int = 1,
numeric_only: _bool = False,
**kwargs: Any,
) -> float: ...
@overload
def var(
self: SupportsGetItem[Scalar, SupportsTruedivInt[S2]],
axis: AxisIndex | None = 0,
skipna: _bool | None = True,
ddof: int = 1,
numeric_only: _bool = False,
**kwargs: Any,
) -> S2: ...
# Rename axis with `mapper`, `axis`, and `inplace=True`
@overload
def rename_axis(
Expand Down
26 changes: 25 additions & 1 deletion tests/series/test_agg.py
View file
Open in desktop
Original file line number Diff line number Diff line change
@@ -1,6 +1,9 @@
import numpy as np
import pandas as pd
from typing_extensions import assert_type
from typing_extensions import (
Never,
assert_type,
)

from tests import (
TYPE_CHECKING_INVALID_USAGE,
Expand All @@ -14,27 +17,31 @@ def test_agg_any_float() -> None:
check(assert_type(series.mean(), float), np.float64)
check(assert_type(series.median(), float), np.float64)
check(assert_type(series.std(), float), np.float64)
check(assert_type(series.var(), float), np.float64)


def test_agg_bool() -> None:
series = pd.Series([True, False, True])
check(assert_type(series.mean(), float), np.float64)
check(assert_type(series.median(), float), np.float64)
check(assert_type(series.std(), float), np.float64)
check(assert_type(series.var(), float), np.float64)


def test_agg_int() -> None:
series = pd.Series([3, 1, 2])
check(assert_type(series.mean(), float), np.float64)
check(assert_type(series.median(), float), np.float64)
check(assert_type(series.std(), float), np.float64)
check(assert_type(series.var(), float), np.float64)


def test_agg_float() -> None:
series = pd.Series([3.0, float("nan"), 2.0])
check(assert_type(series.mean(), float), np.float64)
check(assert_type(series.median(), float), np.float64)
check(assert_type(series.std(), float), np.float64)
check(assert_type(series.var(), float), np.float64)


def test_agg_complex() -> None:
Expand All @@ -57,6 +64,12 @@ def test_agg_complex() -> None:
),
):
check(assert_type(series.std(), np.float64), np.float64)
with pytest_warns_bounded(
np.exceptions.ComplexWarning,
r"Casting complex values to real discards the imaginary part",
upper="2.3.99",
):
check(assert_type(series.var(), float), np.float64)


def test_agg_str() -> None:
Expand All @@ -65,6 +78,7 @@ def test_agg_str() -> None:
series.mean() # type: ignore[misc] # pyright: ignore[reportAttributeAccessIssue]
series.median() # type: ignore[misc] # pyright: ignore[reportAttributeAccessIssue]
series.std() # type: ignore[misc] # pyright: ignore[reportAttributeAccessIssue]
series.var() # type: ignore[misc] # pyright: ignore[reportAttributeAccessIssue]


def test_agg_ts() -> None:
Expand All @@ -75,6 +89,11 @@ def test_agg_ts() -> None:
check(assert_type(series.median(), pd.Timestamp), pd.Timestamp)
check(assert_type(series.std(), pd.Timedelta), pd.Timedelta)

if TYPE_CHECKING_INVALID_USAGE:

def _0() -> None: # pyright: ignore[reportUnusedFunction]
assert_type(series.var(), Never)


def test_agg_td() -> None:
series = pd.Series(pd.to_timedelta(["1 days", "2 days", "3 days"]))
Expand All @@ -83,3 +102,8 @@ def test_agg_td() -> None:
check(assert_type(series.mean(), pd.Timedelta), pd.Timedelta)
check(assert_type(series.median(), pd.Timedelta), pd.Timedelta)
check(assert_type(series.std(), pd.Timedelta), pd.Timedelta)

if TYPE_CHECKING_INVALID_USAGE:

def _0() -> None: # pyright: ignore[reportUnusedFunction]
assert_type(series.var(), Never)
6 changes: 3 additions & 3 deletions tests/series/test_series.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -780,9 +780,9 @@ def test_types_abs() -> None:

def test_types_var() -> None:
s = pd.Series([-10, 2, 3, 10])
s.var()
s.var(axis=0, ddof=1)
s.var(skipna=True, numeric_only=False)
check(assert_type(s.var(), float), np.float64)
check(assert_type(s.var(axis=0, ddof=1), float), np.float64)
check(assert_type(s.var(skipna=True, numeric_only=False), float), np.float64)


def test_types_std() -> None:
Expand Down