Skip to content

Commit 74b0cbf

Browse files
bashtagebashtage
committed
MAINT: Merge main
2 parents ae9251c + 426e31d commit 74b0cbf

File tree

11 files changed

+121
-46
lines changed

11 files changed

+121
-46
lines changed

‎.github/workflows/test.yml

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -22,7 +22,7 @@ jobs:
2222
python-version: ${{ matrix.python-version }}
2323

2424
- name: Install Poetry
25-
run: pip install poetry==1.1.15
25+
run: pip install poetry
2626

2727
- name: Determine poetry version
2828
run: echo "::set-output name=VERSION::$(poetry --version)"

‎pandas-stubs/_typing.pyi

Lines changed: 4 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -176,6 +176,8 @@ IndexingInt = Union[
176176
# passed in, a DataFrame is always returned.
177177
NDFrameT = TypeVar("NDFrameT", bound=NDFrame)
178178

179+
IndexT = TypeVar("IndexT", bound=Index)
180+
179181
# Interval closed type
180182

181183
IntervalClosedType = Literal["left", "right", "both", "neither"]
@@ -246,4 +248,6 @@ ParquetEngine = Literal["auto", "pyarrow", "fastparquet"]
246248
FileWriteMode = Literal[
247249
"a", "w", "x", "at", "wt", "xt", "ab", "wb", "xb", "w+", "w+b", "a+", "a+b"
248250
]
251+
ColspaceArgType = str | int | Sequence[int | str] | Mapping[Hashable, str | int]
252+
249253
__all__ = ["npt", "type_t"]

‎pandas-stubs/core/frame.pyi

Lines changed: 47 additions & 13 deletions
Original file line numberDiff line numberDiff line change
@@ -49,6 +49,7 @@ from pandas._typing import (
4949
Axes,
5050
Axis,
5151
AxisType,
52+
ColspaceArgType,
5253
CompressionOptions,
5354
Dtype,
5455
DtypeNp,
@@ -326,23 +327,39 @@ class DataFrame(NDFrame, OpsMixin):
326327
@overload
327328
def to_html(
328329
self,
329-
buf: FilePathOrBuffer | None,
330-
columns: Sequence[_str] | None = ...,
331-
col_space: int | list[int] | dict[_str | int, int] | None = ...,
330+
buf: FilePath | WriteBuffer[str],
331+
columns: list[HashableT] | None = ...,
332+
col_space: ColspaceArgType | None = ...,
332333
header: _bool = ...,
333334
index: _bool = ...,
334335
na_rep: _str = ...,
335-
formatters=...,
336-
float_format=...,
336+
formatters: list[Callable[[object], str]]
337+
| tuple[Callable[[object], str], ...]
338+
| Mapping[Hashable, Callable[[object], str]]
339+
| None = ...,
340+
float_format: Callable[[float], str] | None = ...,
337341
sparsify: _bool | None = ...,
338342
index_names: _bool = ...,
339-
justify: _str | None = ...,
343+
justify: Literal[
344+
"left",
345+
"right",
346+
"center",
347+
"justify",
348+
"justify-all",
349+
"start",
350+
"end",
351+
"inherit",
352+
"match-parent",
353+
"initial",
354+
"unset",
355+
]
356+
| None = ...,
340357
max_rows: int | None = ...,
341358
max_cols: int | None = ...,
342359
show_dimensions: _bool = ...,
343360
decimal: _str = ...,
344361
bold_rows: _bool = ...,
345-
classes: _str | list | tuple | None = ...,
362+
classes: Sequence[str] | None = ...,
346363
escape: _bool = ...,
347364
notebook: _bool = ...,
348365
border: int | None = ...,
@@ -353,22 +370,39 @@ class DataFrame(NDFrame, OpsMixin):
353370
@overload
354371
def to_html(
355372
self,
356-
columns: Sequence[_str] | None = ...,
357-
col_space: int | list[int] | dict[_str | int, int] | None = ...,
373+
buf: None = ...,
374+
columns: Sequence[HashableT] | None = ...,
375+
col_space: ColspaceArgType | None = ...,
358376
header: _bool = ...,
359377
index: _bool = ...,
360378
na_rep: _str = ...,
361-
formatters=...,
362-
float_format=...,
379+
formatters: list[Callable[[object], str]]
380+
| tuple[Callable[[object], str], ...]
381+
| Mapping[Hashable, Callable[[object], str]]
382+
| None = ...,
383+
float_format: Callable[[float], str] | None = ...,
363384
sparsify: _bool | None = ...,
364385
index_names: _bool = ...,
365-
justify: _str | None = ...,
386+
justify: Literal[
387+
"left",
388+
"right",
389+
"center",
390+
"justify",
391+
"justify-all",
392+
"start",
393+
"end",
394+
"inherit",
395+
"match-parent",
396+
"initial",
397+
"unset",
398+
]
399+
| None = ...,
366400
max_rows: int | None = ...,
367401
max_cols: int | None = ...,
368402
show_dimensions: _bool = ...,
369403
decimal: _str = ...,
370404
bold_rows: _bool = ...,
371-
classes: _str | list | tuple | None = ...,
405+
classes: Sequence[str] | None = ...,
372406
escape: _bool = ...,
373407
notebook: _bool = ...,
374408
border: int | None = ...,

‎pandas-stubs/core/generic.pyi

Lines changed: 3 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -30,6 +30,7 @@ from pandas._typing import (
3030
HDFCompLib,
3131
IgnoreRaise,
3232
Level,
33+
NDFrameT,
3334
ReplaceMethod,
3435
Scalar,
3536
SeriesAxisType,
@@ -71,8 +72,8 @@ class NDFrame(PandasObject, indexing.IndexingMixin):
7172
def pop(self, item: _str) -> NDFrame: ...
7273
def squeeze(self, axis=...): ...
7374
def equals(self, other: Series[S1]) -> _bool: ...
74-
def __neg__(self) -> None: ...
75-
def __pos__(self) -> None: ...
75+
def __neg__(self: NDFrameT) -> NDFrameT: ...
76+
def __pos__(self: NDFrameT) -> NDFrameT: ...
7677
def __nonzero__(self) -> None: ...
7778
def bool(self) -> _bool: ...
7879
def __abs__(self) -> NDFrame: ...

‎pandas-stubs/core/indexes/base.pyi

Lines changed: 3 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -27,6 +27,7 @@ from pandas._typing import (
2727
Dtype,
2828
DtypeArg,
2929
DtypeObj,
30+
IndexT,
3031
Label,
3132
Level,
3233
NaPosition,
@@ -142,11 +143,12 @@ class Index(IndexOpsMixin, PandasObject):
142143
def __and__(self, other) -> Index: ...
143144
def __or__(self, other) -> Index: ...
144145
def __xor__(self, other) -> Index: ...
146+
def __neg__(self: IndexT) -> IndexT: ...
145147
def __nonzero__(self) -> None: ...
146148
__bool__ = ...
147149
def union(self, other: list[T1] | Index, sort=...) -> Index: ...
148150
def intersection(self, other: list[T1] | Index, sort: bool = ...) -> Index: ...
149-
def difference(self, other: list[T1 | None] | Index) -> Index: ...
151+
def difference(self, other: list | Index) -> Index: ...
150152
def symmetric_difference(
151153
self, other: list[T1] | Index, result_name=..., sort=...
152154
) -> Index: ...

‎pandas-stubs/io/html.pyi

Lines changed: 25 additions & 26 deletions
Original file line numberDiff line numberDiff line change
@@ -1,46 +1,45 @@
11
from typing import (
22
Any,
33
Callable,
4-
Iterable,
4+
Hashable,
5+
Literal,
56
Mapping,
7+
Pattern,
68
Sequence,
79
)
810

911
from pandas.core.frame import DataFrame
1012

11-
from pandas._typing import FilePathOrBuffer
12-
13-
class _HtmlFrameParser:
14-
io = ...
15-
match = ...
16-
attrs = ...
17-
encoding = ...
18-
displayed_only = ...
19-
def __init__(self, io, match, attrs, encoding, displayed_only) -> None: ...
20-
def parse_tables(self): ...
21-
22-
class _BeautifulSoupHtml5LibFrameParser(_HtmlFrameParser):
23-
def __init__(self, *args, **kwargs) -> None: ...
24-
25-
class _LxmlFrameParser(_HtmlFrameParser):
26-
def __init__(self, *args, **kwargs) -> None: ...
13+
from pandas._typing import (
14+
FilePath,
15+
HashableT,
16+
ReadBuffer,
17+
)
2718

2819
def read_html(
29-
io: FilePathOrBuffer,
30-
match: str = ...,
20+
io: FilePath | ReadBuffer[str],
21+
match: str | Pattern = ...,
3122
flavor: str | None = ...,
3223
header: int | Sequence[int] | None = ...,
33-
index_col: int | Sequence[Any] | None = ...,
34-
skiprows: int | Sequence[Any] | slice | None = ...,
35-
attrs: Mapping[str, str] | None = ...,
24+
index_col: int | Sequence[int] | list[HashableT] | None = ...,
25+
skiprows: int | Sequence[int] | slice | None = ...,
26+
attrs: dict[str, str] | None = ...,
3627
parse_dates: bool
37-
| Sequence[int | str | Sequence[int | str]]
38-
| dict[str, Sequence[int | str]] = ...,
28+
| Sequence[int]
29+
| list[HashableT] # Cannot be Sequence[Hashable] to prevent str
30+
| Sequence[Sequence[Hashable]]
31+
| dict[str, Sequence[int]]
32+
| dict[str, list[HashableT]] = ...,
3933
thousands: str = ...,
4034
encoding: str | None = ...,
4135
decimal: str = ...,
42-
converters: Mapping[int | str, Callable] | None = ...,
43-
na_values: Iterable[Any] | None = ...,
36+
converters: Mapping[int | HashableT, Callable[[str], Any]] | None = ...,
37+
na_values: str
38+
| list[str]
39+
| dict[HashableT, str]
40+
| dict[HashableT, list[str]]
41+
| None = ...,
4442
keep_default_na: bool = ...,
4543
displayed_only: bool = ...,
44+
extract_links: Literal["header", "footer", "body", "all"] | None = ...,
4645
) -> list[DataFrame]: ...

‎pyproject.toml

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -39,7 +39,7 @@ mypy = ">=0.971"
3939
pyarrow = ">=9.0.0"
4040
pytest = ">=7.1.2"
4141
pyright = ">=1.1.266"
42-
poethepoet = ">=0.13.1"
42+
poethepoet = "0.16.0"
4343
loguru = ">=0.6.0"
4444
pandas = "1.4.3"
4545
typing-extensions = ">=4.2.0"

‎tests/test_frame.py

Lines changed: 12 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -1685,3 +1685,15 @@ def func() -> MyDataFrame[int]:
16851685
return MyDataFrame[int]({"foo": [1, 2, 3]})
16861686

16871687
func()
1688+
1689+
1690+
def test_neg() -> None:
1691+
# GH 253
1692+
df = pd.DataFrame(data={"col1": [1, 2], "col2": [3, 4]})
1693+
check(assert_type(-df, pd.DataFrame), pd.DataFrame)
1694+
1695+
1696+
def test_pos() -> None:
1697+
# GH 253
1698+
df = pd.DataFrame(data={"col1": [1, 2], "col2": [3, 4]})
1699+
check(assert_type(+df, pd.DataFrame), pd.DataFrame)

‎tests/test_indexes.py

Lines changed: 9 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -9,7 +9,6 @@
99

1010

1111
def test_index_unique() -> None:
12-
1312
df = pd.DataFrame({"x": [1, 2, 3, 4]}, index=pd.Index([1, 2, 3, 2]))
1413
ind = df.index
1514
check(assert_type(ind, pd.Index), pd.Index)
@@ -67,7 +66,9 @@ def test_column_contains() -> None:
6766
def test_difference_none() -> None:
6867
# https://github.com/pandas-dev/pandas-stubs/issues/17
6968
ind = pd.Index([1, 2, 3])
70-
check(assert_type(ind.difference([1, None]), "pd.Index"), pd.Index, int)
69+
check(assert_type(ind.difference([1, None]), pd.Index), pd.Index)
70+
# GH 253
71+
check(assert_type(ind.difference([1]), pd.Index), pd.Index)
7172

7273

7374
def test_str_split() -> None:
@@ -87,3 +88,9 @@ def test_index_dropna():
8788

8889
check(assert_type(midx.dropna(how="all"), pd.MultiIndex), pd.MultiIndex)
8990
check(assert_type(midx.dropna(how="any"), pd.MultiIndex), pd.MultiIndex)
91+
92+
93+
def test_index_neg():
94+
# GH 253
95+
idx = pd.Index([1, 2])
96+
check(assert_type(-idx, pd.Index), pd.Index)

‎tests/test_io.py

Lines changed: 8 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -19,6 +19,7 @@
1919
read_feather,
2020
read_fwf,
2121
read_hdf,
22+
read_html,
2223
read_json,
2324
read_orc,
2425
read_parquet,
@@ -438,3 +439,10 @@ def test_to_csv_series():
438439
check(assert_type(s.to_csv(), str), str)
439440
with ensure_clean() as path:
440441
check(assert_type(s.to_csv(path), None), type(None))
442+
443+
444+
def test_read_html():
445+
check(assert_type(DF.to_html(), str), str)
446+
with ensure_clean() as path:
447+
check(assert_type(DF.to_html(path), None), type(None))
448+
check(assert_type(read_html(path), List[DataFrame]), list)

‎tests/test_series.py

Lines changed: 8 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -1124,3 +1124,11 @@ def test_resample() -> None:
11241124
check(assert_type(df.resample("2T").sem(), pd.Series), pd.Series)
11251125
check(assert_type(df.resample("2T").median(), pd.Series), pd.Series)
11261126
check(assert_type(df.resample("2T").ohlc(), pd.DataFrame), pd.DataFrame)
1127+
1128+
1129+
def test_neg() -> None:
1130+
# GH 253
1131+
sr = pd.Series([1, 2, 3])
1132+
sr_int = pd.Series([1, 2, 3], dtype=int)
1133+
check(assert_type(-sr, pd.Series), pd.Series)
1134+
check(assert_type(-sr_int, "pd.Series[int]"), pd.Series, int)

0 commit comments

Comments
 (0)