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_tz_localize.py (3108B)

import pytest
import pytz

from pandas._libs.tslibs import timezones

from pandas import DatetimeIndex, NaT, Series, Timestamp, date_range
import pandas._testing as tm

class TestTZLocalize:
    def test_series_tz_localize(self):

rng = date_range("1/1/2011", periods=100, freq="H")
        ts = Series(1, index=rng)

result = ts.tz_localize("utc")
        assert result.index.tz.zone == "UTC"

# Can't localize if already tz-aware
        rng = date_range("1/1/2011", periods=100, freq="H", tz="utc")
        ts = Series(1, index=rng)

with pytest.raises(TypeError, match="Already tz-aware"):
            ts.tz_localize("US/Eastern")

def test_series_tz_localize_ambiguous_bool(self):
        # make sure that we are correctly accepting bool values as ambiguous

# GH#14402
        ts = Timestamp("2015-11-01 01:00:03")
        expected0 = Timestamp("2015-11-01 01:00:03-0500", tz="US/Central")
        expected1 = Timestamp("2015-11-01 01:00:03-0600", tz="US/Central")

ser = Series([ts])
        expected0 = Series([expected0])
        expected1 = Series([expected1])

with tm.external_error_raised(pytz.AmbiguousTimeError):
            ser.dt.tz_localize("US/Central")

result = ser.dt.tz_localize("US/Central", ambiguous=True)
        tm.assert_series_equal(result, expected0)

result = ser.dt.tz_localize("US/Central", ambiguous=[True])
        tm.assert_series_equal(result, expected0)

result = ser.dt.tz_localize("US/Central", ambiguous=False)
        tm.assert_series_equal(result, expected1)

result = ser.dt.tz_localize("US/Central", ambiguous=[False])
        tm.assert_series_equal(result, expected1)

@pytest.mark.parametrize("tz", ["Europe/Warsaw", "dateutil/Europe/Warsaw"])
    @pytest.mark.parametrize(
        "method, exp",
        [
            ["shift_forward", "2015-03-29 03:00:00"],
            ["NaT", NaT],
            ["raise", None],
            ["foo", "invalid"],
        ],
    )
    def test_series_tz_localize_nonexistent(self, tz, method, exp):
        # GH 8917
        n = 60
        dti = date_range(start="2015-03-29 02:00:00", periods=n, freq="min")
        s = Series(1, dti)
        if method == "raise":
            with tm.external_error_raised(pytz.NonExistentTimeError):
                s.tz_localize(tz, nonexistent=method)
        elif exp == "invalid":
            with pytest.raises(ValueError, match="argument must be one of"):
                dti.tz_localize(tz, nonexistent=method)
        else:
            result = s.tz_localize(tz, nonexistent=method)
            expected = Series(1, index=DatetimeIndex([exp] * n, tz=tz))
            tm.assert_series_equal(result, expected)

@pytest.mark.parametrize("tzstr", ["US/Eastern", "dateutil/US/Eastern"])
    def test_series_tz_localize_empty(self, tzstr):
        # GH#2248
        ser = Series(dtype=object)

ser2 = ser.tz_localize("utc")
        assert ser2.index.tz == pytz.utc

ser2 = ser.tz_localize(tzstr)
        timezones.tz_compare(ser2.index.tz, timezones.maybe_get_tz(tzstr))

import pytest import pytz from pandas._libs.tslibs import timezones from pandas import DatetimeIndex, NaT, Series, Timestamp, date_range import pandas._testing as tm class TestTZLocalize: def test_series_tz_localize(self): rng = date_range("1/1/2011", periods=100, freq="H") ts = Series(1, index=rng) result = ts.tz_localize("utc") assert result.index.tz.zone == "UTC" # Can't localize if already tz-aware rng = date_range("1/1/2011", periods=100, freq="H", tz="utc") ts = Series(1, index=rng) with pytest.raises(TypeError, match="Already tz-aware"): ts.tz_localize("US/Eastern") def test_series_tz_localize_ambiguous_bool(self): # make sure that we are correctly accepting bool values as ambiguous # GH#14402 ts = Timestamp("2015-11-01 01:00:03") expected0 = Timestamp("2015-11-01 01:00:03-0500", tz="US/Central") expected1 = Timestamp("2015-11-01 01:00:03-0600", tz="US/Central") ser = Series([ts]) expected0 = Series([expected0]) expected1 = Series([expected1]) with tm.external_error_raised(pytz.AmbiguousTimeError): ser.dt.tz_localize("US/Central") result = ser.dt.tz_localize("US/Central", ambiguous=True) tm.assert_series_equal(result, expected0) result = ser.dt.tz_localize("US/Central", ambiguous=[True]) tm.assert_series_equal(result, expected0) result = ser.dt.tz_localize("US/Central", ambiguous=False) tm.assert_series_equal(result, expected1) result = ser.dt.tz_localize("US/Central", ambiguous=[False]) tm.assert_series_equal(result, expected1) @pytest.mark.parametrize("tz", ["Europe/Warsaw", "dateutil/Europe/Warsaw"]) @pytest.mark.parametrize( "method, exp", [ ["shift_forward", "2015-03-29 03:00:00"], ["NaT", NaT], ["raise", None], ["foo", "invalid"], ], ) def test_series_tz_localize_nonexistent(self, tz, method, exp): # GH 8917 n = 60 dti = date_range(start="2015-03-29 02:00:00", periods=n, freq="min") s = Series(1, dti) if method == "raise": with tm.external_error_raised(pytz.NonExistentTimeError): s.tz_localize(tz, nonexistent=method) elif exp == "invalid": with pytest.raises(ValueError, match="argument must be one of"): dti.tz_localize(tz, nonexistent=method) else: result = s.tz_localize(tz, nonexistent=method) expected = Series(1, index=DatetimeIndex([exp] * n, tz=tz)) tm.assert_series_equal(result, expected) @pytest.mark.parametrize("tzstr", ["US/Eastern", "dateutil/US/Eastern"]) def test_series_tz_localize_empty(self, tzstr): # GH#2248 ser = Series(dtype=object) ser2 = ser.tz_localize("utc") assert ser2.index.tz == pytz.utc ser2 = ser.tz_localize(tzstr) timezones.tz_compare(ser2.index.tz, timezones.maybe_get_tz(tzstr))