Skip to content

Commit 076f913

Browse files
committed
TST: Move Period truncate test to series/test_period
1 parent a626271 commit 076f913

File tree

2 files changed

+30
-30
lines changed

2 files changed

+30
-30
lines changed

pandas/tests/indexes/period/test_indexing.py

-30
Original file line numberDiff line numberDiff line change
@@ -312,36 +312,6 @@ def test_take_fill_value(self):
312312
with pytest.raises(IndexError):
313313
idx.take(np.array([1, -5]))
314314

315-
def test_truncate(self):
316-
# GH 17717
317-
idx1 = pd.PeriodIndex([
318-
pd.Period('2017-09-02'),
319-
pd.Period('2017-09-02'),
320-
pd.Period('2017-09-03')
321-
])
322-
series1 = pd.Series([1, 2, 3], index=idx1)
323-
result1 = series1.truncate(after='2017-09-02')
324-
325-
expected_idx1 = pd.PeriodIndex([
326-
pd.Period('2017-09-02'),
327-
pd.Period('2017-09-02')
328-
])
329-
tm.assert_series_equal(result1, pd.Series([1, 2], index=expected_idx1))
330-
331-
idx2 = pd.PeriodIndex([
332-
pd.Period('2017-09-03'),
333-
pd.Period('2017-09-02'),
334-
pd.Period('2017-09-03')
335-
])
336-
series2 = pd.Series([1, 2, 3], index=idx2)
337-
result2 = series2.truncate(after='2017-09-02')
338-
339-
expected_idx2 = pd.PeriodIndex([
340-
pd.Period('2017-09-03'),
341-
pd.Period('2017-09-02')
342-
])
343-
tm.assert_series_equal(result2, pd.Series([1, 2], index=expected_idx2))
344-
345315
def test_get_loc(self):
346316
# GH 17717
347317
p1 = pd.Period('2017-09-02')

pandas/tests/series/test_period.py

+30
Original file line numberDiff line numberDiff line change
@@ -249,3 +249,33 @@ def test_align_series(self):
249249
msg = "Input has different freq=D from PeriodIndex\\(freq=A-DEC\\)"
250250
with tm.assert_raises_regex(period.IncompatibleFrequency, msg):
251251
ts + ts.asfreq('D', how="end")
252+
253+
def test_truncate(self):
254+
# GH 17717
255+
idx1 = pd.PeriodIndex([
256+
pd.Period('2017-09-02'),
257+
pd.Period('2017-09-02'),
258+
pd.Period('2017-09-03')
259+
])
260+
series1 = pd.Series([1, 2, 3], index=idx1)
261+
result1 = series1.truncate(after='2017-09-02')
262+
263+
expected_idx1 = pd.PeriodIndex([
264+
pd.Period('2017-09-02'),
265+
pd.Period('2017-09-02')
266+
])
267+
tm.assert_series_equal(result1, pd.Series([1, 2], index=expected_idx1))
268+
269+
idx2 = pd.PeriodIndex([
270+
pd.Period('2017-09-03'),
271+
pd.Period('2017-09-02'),
272+
pd.Period('2017-09-03')
273+
])
274+
series2 = pd.Series([1, 2, 3], index=idx2)
275+
result2 = series2.truncate(after='2017-09-02')
276+
277+
expected_idx2 = pd.PeriodIndex([
278+
pd.Period('2017-09-03'),
279+
pd.Period('2017-09-02')
280+
])
281+
tm.assert_series_equal(result2, pd.Series([1, 2], index=expected_idx2))

0 commit comments

Comments
 (0)