Skip to content

Commit ce546ff

Browse files
authored
pandas 2.0 removals in testing and clipboard (#933)
* Fix CI from PR 930 * More pandas 2.0 removals * Revert "Fix CI from PR 930" This reverts commit 6ecbab9. * Fix function signature * Fix all functions in pandas.testing
1 parent 2acc94d commit ce546ff

File tree

2 files changed

+95
-52
lines changed

2 files changed

+95
-52
lines changed

pandas-stubs/_testing/__init__.pyi

+95-46
Original file line numberDiff line numberDiff line change
@@ -1,65 +1,117 @@
1-
from collections.abc import Generator
1+
from collections.abc import (
2+
Container,
3+
Generator,
4+
Iterable,
5+
)
26
from contextlib import contextmanager
37
from typing import (
48
Any,
59
Literal,
610
overload,
711
)
12+
import warnings
813

14+
from matplotlib.artist import Artist
15+
import numpy as np
916
from pandas import (
17+
Categorical,
1018
DataFrame,
1119
Index,
1220
Series,
1321
)
22+
from pandas.arrays import (
23+
DatetimeArray,
24+
IntervalArray,
25+
PeriodArray,
26+
SparseArray,
27+
TimedeltaArray,
28+
)
29+
from pandas.core.arrays.base import ExtensionArray
30+
31+
from pandas._typing import (
32+
AnyArrayLike,
33+
T,
34+
)
1435

1536
def assert_almost_equal(
16-
left,
17-
right,
18-
check_dtype: bool | str = ...,
19-
check_less_precise: bool | int = ...,
37+
left: T,
38+
right: T,
39+
check_dtype: bool | Literal["equiv"] = ...,
40+
rtol: float = ...,
41+
atol: float = ...,
2042
**kwargs,
21-
): ...
22-
def assert_dict_equal(left, right, compare_keys: bool = ...): ...
23-
def assert_index_equal(left: Index, right: Index) -> None: ...
24-
def assert_class_equal(left, right, exact: bool | str = ..., obj=...): ...
25-
def assert_attr_equal(attr, left, right, obj: str = ...): ...
26-
def assert_is_valid_plot_return_object(objs) -> None: ...
27-
def assert_is_sorted(seq) -> None: ...
43+
) -> None: ...
44+
def assert_dict_equal(left: dict, right: dict, compare_keys: bool = ...) -> None: ...
45+
def assert_index_equal(
46+
left: Index,
47+
right: Index,
48+
exact: bool | Literal["equiv"] = ...,
49+
check_names: bool = ...,
50+
check_exact: bool = ...,
51+
check_categorical: bool = ...,
52+
check_order: bool = ...,
53+
rtol: float = ...,
54+
atol: float = ...,
55+
obj: str = ...,
56+
) -> None: ...
57+
def assert_class_equal(
58+
left: T, right: T, exact: bool | Literal["equiv"] = ..., obj: str = ...
59+
) -> None: ...
60+
def assert_attr_equal(
61+
attr: str, left: object, right: object, obj: str = ...
62+
) -> None: ...
63+
def assert_is_valid_plot_return_object(
64+
objs: Series | np.ndarray | Artist | tuple | dict,
65+
) -> None: ...
66+
def assert_is_sorted(seq: AnyArrayLike) -> None: ...
2867
def assert_categorical_equal(
29-
left,
30-
right,
68+
left: Categorical,
69+
right: Categorical,
3170
check_dtype: bool = ...,
3271
check_category_order: bool = ...,
3372
obj: str = ...,
3473
) -> None: ...
3574
def assert_interval_array_equal(
36-
left, right, exact: str = ..., obj: str = ...
75+
left: IntervalArray,
76+
right: IntervalArray,
77+
exact: bool | Literal["equiv"] = ...,
78+
obj: str = ...,
79+
) -> None: ...
80+
def assert_period_array_equal(
81+
left: PeriodArray, right: PeriodArray, obj: str = ...
82+
) -> None: ...
83+
def assert_datetime_array_equal(
84+
left: DatetimeArray, right: DatetimeArray, check_freq: bool = ...
85+
) -> None: ...
86+
def assert_timedelta_array_equal(
87+
left: TimedeltaArray, right: TimedeltaArray, check_freq: bool = ...
3788
) -> None: ...
38-
def assert_period_array_equal(left, right, obj: str = ...) -> None: ...
39-
def assert_datetime_array_equal(left, right, obj: str = ...) -> None: ...
40-
def assert_timedelta_array_equal(left, right, obj: str = ...) -> None: ...
4189
def assert_numpy_array_equal(
4290
left,
4391
right,
4492
strict_nan: bool = ...,
45-
check_dtype: bool = ...,
46-
err_msg=...,
47-
check_same=...,
93+
check_dtype: bool | Literal["equiv"] = ...,
94+
err_msg: str | None = ...,
95+
check_same: Literal["copy", "same"] | None = ...,
4896
obj: str = ...,
49-
): ...
97+
index_values: Index | np.ndarray | None = ...,
98+
) -> None: ...
5099
def assert_extension_array_equal(
51-
left,
52-
right,
53-
check_dtype: bool = ...,
54-
check_less_precise: bool = ...,
100+
left: ExtensionArray,
101+
right: ExtensionArray,
102+
check_dtype: bool | Literal["equiv"] = ...,
103+
index_values: Index | np.ndarray | None = ...,
55104
check_exact: bool = ...,
105+
rtol: float = ...,
106+
atol: float = ...,
107+
obj: str = ...,
56108
) -> None: ...
57109
@overload
58110
def assert_series_equal(
59111
left: Series,
60112
right: Series,
61-
check_dtype: bool = ...,
62-
check_index_type: bool | str = ...,
113+
check_dtype: bool | Literal["equiv"] = ...,
114+
check_index_type: bool | Literal["equiv"] = ...,
63115
check_series_type: bool = ...,
64116
check_names: bool = ...,
65117
check_exact: bool = ...,
@@ -79,8 +131,8 @@ def assert_series_equal(
79131
def assert_series_equal(
80132
left: Series,
81133
right: Series,
82-
check_dtype: bool = ...,
83-
check_index_type: bool | str = ...,
134+
check_dtype: bool | Literal["equiv"] = ...,
135+
check_index_type: bool | Literal["equiv"] = ...,
84136
check_series_type: bool = ...,
85137
check_names: bool = ...,
86138
check_exact: bool = ...,
@@ -103,37 +155,34 @@ def assert_frame_equal(
103155
check_index_type: bool | Literal["equiv"] = ...,
104156
check_column_type: bool | Literal["equiv"] = ...,
105157
check_frame_type: bool = ...,
106-
check_less_precise: int | bool = ...,
107158
check_names: bool = ...,
108159
by_blocks: bool = ...,
109160
check_exact: bool = ...,
110161
check_datetimelike_compat: bool = ...,
111162
check_categorical: bool = ...,
112163
check_like: bool = ...,
113164
check_freq: bool = ...,
114-
check_flag: bool = ...,
165+
check_flags: bool = ...,
115166
rtol: float = ...,
116167
atol: float = ...,
117168
obj: str = ...,
118169
) -> None: ...
119170
def assert_equal(left, right, **kwargs) -> None: ...
120-
def assert_sp_array_equal(
121-
left,
122-
right,
123-
check_dtype: bool = ...,
124-
check_kind: bool = ...,
125-
check_fill_value: bool = ...,
126-
consolidate_block_indices: bool = ...,
127-
) -> None: ...
128-
def assert_contains_all(iterable, dic) -> None: ...
129-
def assert_copy(iter1, iter2, **eql_kwargs) -> None: ...
171+
def assert_sp_array_equal(left: SparseArray, right: SparseArray) -> None: ...
172+
def assert_contains_all(iterable: Iterable[T], dic: Container[T]) -> None: ...
173+
def assert_copy(iter1: Iterable[T], iter2: Iterable[T], **eql_kwargs) -> None: ...
174+
@contextmanager
130175
def assert_produces_warning(
131-
expected_warning=...,
132-
filter_level: str = ...,
133-
clear=...,
176+
expected_warning: (
177+
type[Warning] | Literal[False] | tuple[type[Warning], ...] | None
178+
) = ...,
179+
filter_level: Literal[
180+
"error", "ignore", "always", "default", "module", "once"
181+
] = ...,
134182
check_stacklevel: bool = ...,
135183
raise_on_extra_warnings: bool = ...,
136-
) -> None: ...
184+
match: str | None = None,
185+
) -> Generator[list[warnings.WarningMessage], None, None]: ...
137186
@contextmanager
138187
def ensure_clean(
139188
filename: str | None = ..., return_filelike: bool = ..., **kwargs: Any

pandas-stubs/io/clipboards.pyi

-6
Original file line numberDiff line numberDiff line change
@@ -76,8 +76,6 @@ def read_clipboard(
7676
encoding: str | None = ...,
7777
encoding_errors: str | None = ...,
7878
dialect: str | csv.Dialect = ...,
79-
# error_bad_lines: bool | None = ..., # Deprecated: 1.3.0
80-
# warn_bad_lines: bool | None = ..., # Deprecated: 1.3.0
8179
on_bad_lines: (
8280
Literal["error", "warn", "skip"] | Callable[[list[str]], list[str] | None]
8381
) = ...,
@@ -137,8 +135,6 @@ def read_clipboard(
137135
encoding: str | None = ...,
138136
encoding_errors: str | None = ...,
139137
dialect: str | csv.Dialect = ...,
140-
# error_bad_lines: bool | None = ..., # Deprecated: 1.3.0
141-
# warn_bad_lines: bool | None = ..., # Deprecated: 1.3.0
142138
on_bad_lines: (
143139
Literal["error", "warn", "skip"] | Callable[[list[str]], list[str] | None]
144140
) = ...,
@@ -198,8 +194,6 @@ def read_clipboard(
198194
encoding: str | None = ...,
199195
encoding_errors: str | None = ...,
200196
dialect: str | csv.Dialect = ...,
201-
# error_bad_lines: bool | None = ..., # Deprecated: 1.3.0
202-
# warn_bad_lines: bool | None = ..., # Deprecated: 1.3.0
203197
on_bad_lines: (
204198
Literal["error", "warn", "skip"] | Callable[[list[str]], list[str] | None]
205199
) = ...,

0 commit comments

Comments
 (0)