forked from pandas-dev/pandas-stubs
-
Notifications
You must be signed in to change notification settings - Fork 0
Expand file tree
/
Copy pathtest_sub.py
More file actions
20 lines (14 loc) · 779 Bytes
/
Copy pathtest_sub.py
File metadata and controls
20 lines (14 loc) · 779 Bytes
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
"""Test module for subtraction operation on Series[Period]."""
import pandas as pd
from typing_extensions import assert_type
from pandas._libs.tslibs.offsets import BaseOffset
from tests import check
def test_sub() -> None:
"""Test sub method for pd.Series[pd.Period]."""
p = pd.Period("2012-1-1", freq="D")
sr = pd.Series([pd.Period("2012-1-1", freq="D")])
check(assert_type(sr - sr, "pd.Series[BaseOffset]"), pd.Series, BaseOffset)
check(assert_type(p - sr, "pd.Series[BaseOffset]"), pd.Series, BaseOffset)
check(assert_type(sr - p, "pd.Series[BaseOffset]"), pd.Series, BaseOffset)
check(assert_type(sr.sub(p), "pd.Series[BaseOffset]"), pd.Series, BaseOffset)
check(assert_type(sr.rsub(p), "pd.Series[BaseOffset]"), pd.Series, BaseOffset)