

def test_truncate(self):
    offset = datetools.bday
    ts = self.ts[::3]
    (start, end) = (self.ts.index[3], self.ts.index[6])
    (start_missing, end_missing) = (self.ts.index[2], self.ts.index[7])
    truncated = ts.truncate()
    assert_series_equal(truncated, ts)
    expected = ts[1:3]
    truncated = ts.truncate(start, end)
    assert_series_equal(truncated, expected)
    truncated = ts.truncate(start_missing, end_missing)
    assert_series_equal(truncated, expected)
    expected = ts[1:]
    truncated = ts.truncate(before=start)
    assert_series_equal(truncated, expected)
    truncated = ts.truncate(before=start_missing)
    assert_series_equal(truncated, expected)
    expected = ts[:3]
    truncated = ts.truncate(after=end)
    assert_series_equal(truncated, expected)
    truncated = ts.truncate(after=end_missing)
    assert_series_equal(truncated, expected)
    truncated = ts.truncate(after=(self.ts.index[0] - offset))
    assert (len(truncated) == 0)
    truncated = ts.truncate(before=(self.ts.index[(- 1)] + offset))
    assert (len(truncated) == 0)
    self.assertRaises(ValueError, ts.truncate, before=(self.ts.index[(- 1)] + offset), after=(self.ts.index[0] - offset))

dataset/ETHPy150Open pydata/pandas/pandas/tests/series/test_timeseries.py TestSeriesTimeSeries.test_truncate/original