Skip to content

Commit dbe2179

Browse files
authored
for #410 (#413)
* for #410 added series in astypr * up * up2
1 parent 5df1c51 commit dbe2179

File tree

2 files changed

+7
-1
lines changed

2 files changed

+7
-1
lines changed

pandas-stubs/core/frame.pyi

+1-1
Original file line numberDiff line numberDiff line change
@@ -1359,7 +1359,7 @@ class DataFrame(NDFrame, OpsMixin):
13591359
) -> DataFrame: ...
13601360
def astype(
13611361
self,
1362-
dtype: _str | Dtype | dict[_str, _str | Dtype],
1362+
dtype: _str | Dtype | dict[_str, _str | Dtype] | Series,
13631363
copy: _bool = ...,
13641364
errors: _str = ...,
13651365
) -> DataFrame: ...

tests/test_frame.py

+6
Original file line numberDiff line numberDiff line change
@@ -1921,3 +1921,9 @@ def df_accepting_dicts_iterator() -> None:
19211921
# GH 392
19221922
data = [{"a": 1, "b": 2}, {"a": 3, "b": 5}]
19231923
check(assert_type(pd.DataFrame(iter(data)), pd.DataFrame), pd.DataFrame)
1924+
1925+
1926+
def series_added_in_astype() -> None:
1927+
# GH410
1928+
df = pd.DataFrame({"a": [1, 2, 3], "b": [4, 5, 6]})
1929+
check(assert_type(df.astype(df.dtypes), pd.DataFrame), pd.DataFrame)

0 commit comments

Comments
 (0)