FM / methods

/usr/local/lib64/python3.6/site-packages/pandas/tests/series/methods/

/usr/local/lib64/python3.6/site-packages/pandas/tests/series/methods

Name	Size	Mode	Actions
__pycache__/	-	0755	rm
test_align.py	5359	0644	edit dl rm
test_append.py	9600	0644	edit dl rm
test_argsort.py	2249	0644	edit dl rm
test_asfreq.py	3616	0644	edit dl rm
test_asof.py	5515	0644	edit dl rm
test_astype.py	2628	0644	edit dl rm
test_at_time.py	2629	0644	edit dl rm
test_autocorr.py	999	0644	edit dl rm
test_between.py	1197	0644	edit dl rm
test_between_time.py	5086	0644	edit dl rm
test_clip.py	3384	0644	edit dl rm
test_combine.py	627	0644	edit dl rm
test_combine_first.py	3411	0644	edit dl rm
test_compare.py	3734	0644	edit dl rm
test_convert_dtypes.py	9914	0644	edit dl rm
test_count.py	1123	0644	edit dl rm
test_cov_corr.py	5215	0644	edit dl rm
test_describe.py	4640	0644	edit dl rm
test_diff.py	2331	0644	edit dl rm
test_drop.py	2986	0644	edit dl rm
test_droplevel.py	629	0644	edit dl rm
test_drop_duplicates.py	6850	0644	edit dl rm
test_duplicated.py	996	0644	edit dl rm
test_equals.py	1295	0644	edit dl rm
test_explode.py	3538	0644	edit dl rm
test_fillna.py	6513	0644	edit dl rm
test_first_and_last.py	2042	0644	edit dl rm
test_head_tail.py	343	0644	edit dl rm
test_interpolate.py	27757	0644	edit dl rm
test_isin.py	3445	0644	edit dl rm
test_nlargest.py	7121	0644	edit dl rm
test_pct_change.py	2968	0644	edit dl rm
test_quantile.py	6810	0644	edit dl rm
test_rank.py	20287	0644	edit dl rm
test_reindex_like.py	1245	0644	edit dl rm
test_rename.py	3378	0644	edit dl rm
test_rename_axis.py	1503	0644	edit dl rm
test_replace.py	15559	0644	edit dl rm
test_reset_index.py	4636	0644	edit dl rm
test_round.py	1970	0644	edit dl rm
test_searchsorted.py	2121	0644	edit dl rm
test_shift.py	12595	0644	edit dl rm
test_sort_index.py	12277	0644	edit dl rm
test_sort_values.py	7981	0644	edit dl rm
test_to_dict.py	742	0644	edit dl rm
test_to_period.py	1755	0644	edit dl rm
test_to_timestamp.py	2616	0644	edit dl rm
test_truncate.py	5557	0644	edit dl rm
test_tz_convert.py	1008	0644	edit dl rm
test_tz_localize.py	3108	0644	edit dl rm
test_unstack.py	4208	0644	edit dl rm
test_update.py	4213	0644	edit dl rm
test_value_counts.py	8136	0644	edit dl rm
__init__.py	225	0644	edit dl rm

Edit: /usr/local/lib64/python3.6/site-packages/pandas/tests/series/methods/test_truncate.py (5557B)

from datetime import datetime

import numpy as np
import pytest

import pandas as pd
from pandas import Series, date_range
import pandas._testing as tm

from pandas.tseries.offsets import BDay

class TestTruncate:
    def test_truncate(self, datetime_series):
        offset = BDay()

ts = datetime_series[::3]

start, end = datetime_series.index[3], datetime_series.index[6]
        start_missing, end_missing = datetime_series.index[2], datetime_series.index[7]

# neither specified
        truncated = ts.truncate()
        tm.assert_series_equal(truncated, ts)

# both specified
        expected = ts[1:3]

truncated = ts.truncate(start, end)
        tm.assert_series_equal(truncated, expected)

truncated = ts.truncate(start_missing, end_missing)
        tm.assert_series_equal(truncated, expected)

# start specified
        expected = ts[1:]

truncated = ts.truncate(before=start)
        tm.assert_series_equal(truncated, expected)

truncated = ts.truncate(before=start_missing)
        tm.assert_series_equal(truncated, expected)

# end specified
        expected = ts[:3]

truncated = ts.truncate(after=end)
        tm.assert_series_equal(truncated, expected)

truncated = ts.truncate(after=end_missing)
        tm.assert_series_equal(truncated, expected)

# corner case, empty series returned
        truncated = ts.truncate(after=datetime_series.index[0] - offset)
        assert len(truncated) == 0

truncated = ts.truncate(before=datetime_series.index[-1] + offset)
        assert len(truncated) == 0

msg = "Truncate: 1999-12-31 00:00:00 must be after 2000-02-14 00:00:00"
        with pytest.raises(ValueError, match=msg):
            ts.truncate(
                before=datetime_series.index[-1] + offset,
                after=datetime_series.index[0] - offset,
            )

def test_truncate_nonsortedindex(self):
        # GH#17935

s = pd.Series(["a", "b", "c", "d", "e"], index=[5, 3, 2, 9, 0])
        msg = "truncate requires a sorted index"

with pytest.raises(ValueError, match=msg):
            s.truncate(before=3, after=9)

rng = pd.date_range("2011-01-01", "2012-01-01", freq="W")
        ts = pd.Series(np.random.randn(len(rng)), index=rng)
        msg = "truncate requires a sorted index"

with pytest.raises(ValueError, match=msg):
            ts.sort_values(ascending=False).truncate(before="2011-11", after="2011-12")

@pytest.mark.parametrize(
        "before, after, indices",
        [(1, 2, [2, 1]), (None, 2, [2, 1, 0]), (1, None, [3, 2, 1])],
    )
    @pytest.mark.parametrize("klass", [pd.Int64Index, pd.DatetimeIndex])
    def test_truncate_decreasing_index(self, before, after, indices, klass):
        # https://github.com/pandas-dev/pandas/issues/33756
        idx = klass([3, 2, 1, 0])
        if klass is pd.DatetimeIndex:
            before = pd.Timestamp(before) if before is not None else None
            after = pd.Timestamp(after) if after is not None else None
            indices = [pd.Timestamp(i) for i in indices]
        values = pd.Series(range(len(idx)), index=idx)
        result = values.truncate(before=before, after=after)
        expected = values.loc[indices]
        tm.assert_series_equal(result, expected)

def test_truncate_datetimeindex_tz(self):
        # GH 9243
        idx = date_range("4/1/2005", "4/30/2005", freq="D", tz="US/Pacific")
        s = Series(range(len(idx)), index=idx)
        result = s.truncate(datetime(2005, 4, 2), datetime(2005, 4, 4))
        expected = Series([1, 2, 3], index=idx[1:4])
        tm.assert_series_equal(result, expected)

def test_truncate_periodindex(self):
        # GH 17717
        idx1 = pd.PeriodIndex(
            [pd.Period("2017-09-02"), pd.Period("2017-09-02"), pd.Period("2017-09-03")]
        )
        series1 = pd.Series([1, 2, 3], index=idx1)
        result1 = series1.truncate(after="2017-09-02")

expected_idx1 = pd.PeriodIndex(
            [pd.Period("2017-09-02"), pd.Period("2017-09-02")]
        )
        tm.assert_series_equal(result1, pd.Series([1, 2], index=expected_idx1))

idx2 = pd.PeriodIndex(
            [pd.Period("2017-09-03"), pd.Period("2017-09-02"), pd.Period("2017-09-03")]
        )
        series2 = pd.Series([1, 2, 3], index=idx2)
        result2 = series2.sort_index().truncate(after="2017-09-02")

expected_idx2 = pd.PeriodIndex([pd.Period("2017-09-02")])
        tm.assert_series_equal(result2, pd.Series([2], index=expected_idx2))

def test_truncate_multiindex(self):
        # GH 34564
        mi = pd.MultiIndex.from_product([[1, 2, 3, 4], ["A", "B"]], names=["L1", "L2"])
        s1 = pd.Series(range(mi.shape[0]), index=mi, name="col")
        result = s1.truncate(before=2, after=3)

df = pd.DataFrame.from_dict(
            {"L1": [2, 2, 3, 3], "L2": ["A", "B", "A", "B"], "col": [2, 3, 4, 5]}
        )
        return_value = df.set_index(["L1", "L2"], inplace=True)
        assert return_value is None
        expected = df.col

tm.assert_series_equal(result, expected)

def test_truncate_one_element_series(self):
        # GH 35544
        series = pd.Series([0.1], index=pd.DatetimeIndex(["2020-08-04"]))
        before = pd.Timestamp("2020-08-02")
        after = pd.Timestamp("2020-08-04")

result = series.truncate(before=before, after=after)

# the input Series and the expected Series are the same
        tm.assert_series_equal(result, series)

from datetime import datetime import numpy as np import pytest import pandas as pd from pandas import Series, date_range import pandas._testing as tm from pandas.tseries.offsets import BDay class TestTruncate: def test_truncate(self, datetime_series): offset = BDay() ts = datetime_series[::3] start, end = datetime_series.index[3], datetime_series.index[6] start_missing, end_missing = datetime_series.index[2], datetime_series.index[7] # neither specified truncated = ts.truncate() tm.assert_series_equal(truncated, ts) # both specified expected = ts[1:3] truncated = ts.truncate(start, end) tm.assert_series_equal(truncated, expected) truncated = ts.truncate(start_missing, end_missing) tm.assert_series_equal(truncated, expected) # start specified expected = ts[1:] truncated = ts.truncate(before=start) tm.assert_series_equal(truncated, expected) truncated = ts.truncate(before=start_missing) tm.assert_series_equal(truncated, expected) # end specified expected = ts[:3] truncated = ts.truncate(after=end) tm.assert_series_equal(truncated, expected) truncated = ts.truncate(after=end_missing) tm.assert_series_equal(truncated, expected) # corner case, empty series returned truncated = ts.truncate(after=datetime_series.index[0] - offset) assert len(truncated) == 0 truncated = ts.truncate(before=datetime_series.index[-1] + offset) assert len(truncated) == 0 msg = "Truncate: 1999-12-31 00:00:00 must be after 2000-02-14 00:00:00" with pytest.raises(ValueError, match=msg): ts.truncate( before=datetime_series.index[-1] + offset, after=datetime_series.index[0] - offset, ) def test_truncate_nonsortedindex(self): # GH#17935 s = pd.Series(["a", "b", "c", "d", "e"], index=[5, 3, 2, 9, 0]) msg = "truncate requires a sorted index" with pytest.raises(ValueError, match=msg): s.truncate(before=3, after=9) rng = pd.date_range("2011-01-01", "2012-01-01", freq="W") ts = pd.Series(np.random.randn(len(rng)), index=rng) msg = "truncate requires a sorted index" with pytest.raises(ValueError, match=msg): ts.sort_values(ascending=False).truncate(before="2011-11", after="2011-12") @pytest.mark.parametrize( "before, after, indices", [(1, 2, [2, 1]), (None, 2, [2, 1, 0]), (1, None, [3, 2, 1])], ) @pytest.mark.parametrize("klass", [pd.Int64Index, pd.DatetimeIndex]) def test_truncate_decreasing_index(self, before, after, indices, klass): # https://github.com/pandas-dev/pandas/issues/33756 idx = klass([3, 2, 1, 0]) if klass is pd.DatetimeIndex: before = pd.Timestamp(before) if before is not None else None after = pd.Timestamp(after) if after is not None else None indices = [pd.Timestamp(i) for i in indices] values = pd.Series(range(len(idx)), index=idx) result = values.truncate(before=before, after=after) expected = values.loc[indices] tm.assert_series_equal(result, expected) def test_truncate_datetimeindex_tz(self): # GH 9243 idx = date_range("4/1/2005", "4/30/2005", freq="D", tz="US/Pacific") s = Series(range(len(idx)), index=idx) result = s.truncate(datetime(2005, 4, 2), datetime(2005, 4, 4)) expected = Series([1, 2, 3], index=idx[1:4]) tm.assert_series_equal(result, expected) def test_truncate_periodindex(self): # GH 17717 idx1 = pd.PeriodIndex( [pd.Period("2017-09-02"), pd.Period("2017-09-02"), pd.Period("2017-09-03")] ) series1 = pd.Series([1, 2, 3], index=idx1) result1 = series1.truncate(after="2017-09-02") expected_idx1 = pd.PeriodIndex( [pd.Period("2017-09-02"), pd.Period("2017-09-02")] ) tm.assert_series_equal(result1, pd.Series([1, 2], index=expected_idx1)) idx2 = pd.PeriodIndex( [pd.Period("2017-09-03"), pd.Period("2017-09-02"), pd.Period("2017-09-03")] ) series2 = pd.Series([1, 2, 3], index=idx2) result2 = series2.sort_index().truncate(after="2017-09-02") expected_idx2 = pd.PeriodIndex([pd.Period("2017-09-02")]) tm.assert_series_equal(result2, pd.Series([2], index=expected_idx2)) def test_truncate_multiindex(self): # GH 34564 mi = pd.MultiIndex.from_product([[1, 2, 3, 4], ["A", "B"]], names=["L1", "L2"]) s1 = pd.Series(range(mi.shape[0]), index=mi, name="col") result = s1.truncate(before=2, after=3) df = pd.DataFrame.from_dict( {"L1": [2, 2, 3, 3], "L2": ["A", "B", "A", "B"], "col": [2, 3, 4, 5]} ) return_value = df.set_index(["L1", "L2"], inplace=True) assert return_value is None expected = df.col tm.assert_series_equal(result, expected) def test_truncate_one_element_series(self): # GH 35544 series = pd.Series([0.1], index=pd.DatetimeIndex(["2020-08-04"])) before = pd.Timestamp("2020-08-02") after = pd.Timestamp("2020-08-04") result = series.truncate(before=before, after=after) # the input Series and the expected Series are the same tm.assert_series_equal(result, series)