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_reset_index.py (4636B)

import numpy as np
import pytest

import pandas as pd
from pandas import DataFrame, Index, MultiIndex, RangeIndex, Series
import pandas._testing as tm

class TestResetIndex:
    def test_reset_index(self):
        df = tm.makeDataFrame()[:5]
        ser = df.stack()
        ser.index.names = ["hash", "category"]

ser.name = "value"
        df = ser.reset_index()
        assert "value" in df

df = ser.reset_index(name="value2")
        assert "value2" in df

# check inplace
        s = ser.reset_index(drop=True)
        s2 = ser
        return_value = s2.reset_index(drop=True, inplace=True)
        assert return_value is None
        tm.assert_series_equal(s, s2)

# level
        index = MultiIndex(
            levels=[["bar"], ["one", "two", "three"], [0, 1]],
            codes=[[0, 0, 0, 0, 0, 0], [0, 1, 2, 0, 1, 2], [0, 1, 0, 1, 0, 1]],
        )
        s = Series(np.random.randn(6), index=index)
        rs = s.reset_index(level=1)
        assert len(rs.columns) == 2

rs = s.reset_index(level=[0, 2], drop=True)
        tm.assert_index_equal(rs.index, Index(index.get_level_values(1)))
        assert isinstance(rs, Series)

def test_reset_index_name(self):
        s = Series([1, 2, 3], index=Index(range(3), name="x"))
        assert s.reset_index().index.name is None
        assert s.reset_index(drop=True).index.name is None

def test_reset_index_level(self):
        df = DataFrame([[1, 2, 3], [4, 5, 6]], columns=["A", "B", "C"])

for levels in ["A", "B"], [0, 1]:
            # With MultiIndex
            s = df.set_index(["A", "B"])["C"]

result = s.reset_index(level=levels[0])
            tm.assert_frame_equal(result, df.set_index("B"))

result = s.reset_index(level=levels[:1])
            tm.assert_frame_equal(result, df.set_index("B"))

result = s.reset_index(level=levels)
            tm.assert_frame_equal(result, df)

result = df.set_index(["A", "B"]).reset_index(level=levels, drop=True)
            tm.assert_frame_equal(result, df[["C"]])

with pytest.raises(KeyError, match="Level E "):
                s.reset_index(level=["A", "E"])

# With single-level Index
            s = df.set_index("A")["B"]

result = s.reset_index(level=levels[0])
            tm.assert_frame_equal(result, df[["A", "B"]])

result = s.reset_index(level=levels[:1])
            tm.assert_frame_equal(result, df[["A", "B"]])

result = s.reset_index(level=levels[0], drop=True)
            tm.assert_series_equal(result, df["B"])

with pytest.raises(IndexError, match="Too many levels"):
                s.reset_index(level=[0, 1, 2])

# Check that .reset_index([],drop=True) doesn't fail
        result = Series(range(4)).reset_index([], drop=True)
        expected = Series(range(4))
        tm.assert_series_equal(result, expected)

def test_reset_index_range(self):
        # GH 12071
        s = Series(range(2), name="A", dtype="int64")
        series_result = s.reset_index()
        assert isinstance(series_result.index, RangeIndex)
        series_expected = DataFrame(
            [[0, 0], [1, 1]], columns=["index", "A"], index=RangeIndex(stop=2)
        )
        tm.assert_frame_equal(series_result, series_expected)

def test_reset_index_drop_errors(self):
        #  GH 20925

# KeyError raised for series index when passed level name is missing
        s = Series(range(4))
        with pytest.raises(KeyError, match="does not match index name"):
            s.reset_index("wrong", drop=True)
        with pytest.raises(KeyError, match="does not match index name"):
            s.reset_index("wrong")

# KeyError raised for series when level to be dropped is missing
        s = Series(range(4), index=MultiIndex.from_product([[1, 2]] * 2))
        with pytest.raises(KeyError, match="not found"):
            s.reset_index("wrong", drop=True)

@pytest.mark.parametrize(
    "array, dtype",
    [
        (["a", "b"], object),
        (
            pd.period_range("12-1-2000", periods=2, freq="Q-DEC"),
            pd.PeriodDtype(freq="Q-DEC"),
        ),
    ],
)
def test_reset_index_dtypes_on_empty_series_with_multiindex(array, dtype):
    # GH 19602 - Preserve dtype on empty Series with MultiIndex
    idx = MultiIndex.from_product([[0, 1], [0.5, 1.0], array])
    result = Series(dtype=object, index=idx)[:0].reset_index().dtypes
    expected = Series(
        {"level_0": np.int64, "level_1": np.float64, "level_2": dtype, 0: object}
    )
    tm.assert_series_equal(result, expected)

import numpy as np import pytest import pandas as pd from pandas import DataFrame, Index, MultiIndex, RangeIndex, Series import pandas._testing as tm class TestResetIndex: def test_reset_index(self): df = tm.makeDataFrame()[:5] ser = df.stack() ser.index.names = ["hash", "category"] ser.name = "value" df = ser.reset_index() assert "value" in df df = ser.reset_index(name="value2") assert "value2" in df # check inplace s = ser.reset_index(drop=True) s2 = ser return_value = s2.reset_index(drop=True, inplace=True) assert return_value is None tm.assert_series_equal(s, s2) # level index = MultiIndex( levels=[["bar"], ["one", "two", "three"], [0, 1]], codes=[[0, 0, 0, 0, 0, 0], [0, 1, 2, 0, 1, 2], [0, 1, 0, 1, 0, 1]], ) s = Series(np.random.randn(6), index=index) rs = s.reset_index(level=1) assert len(rs.columns) == 2 rs = s.reset_index(level=[0, 2], drop=True) tm.assert_index_equal(rs.index, Index(index.get_level_values(1))) assert isinstance(rs, Series) def test_reset_index_name(self): s = Series([1, 2, 3], index=Index(range(3), name="x")) assert s.reset_index().index.name is None assert s.reset_index(drop=True).index.name is None def test_reset_index_level(self): df = DataFrame([[1, 2, 3], [4, 5, 6]], columns=["A", "B", "C"]) for levels in ["A", "B"], [0, 1]: # With MultiIndex s = df.set_index(["A", "B"])["C"] result = s.reset_index(level=levels[0]) tm.assert_frame_equal(result, df.set_index("B")) result = s.reset_index(level=levels[:1]) tm.assert_frame_equal(result, df.set_index("B")) result = s.reset_index(level=levels) tm.assert_frame_equal(result, df) result = df.set_index(["A", "B"]).reset_index(level=levels, drop=True) tm.assert_frame_equal(result, df[["C"]]) with pytest.raises(KeyError, match="Level E "): s.reset_index(level=["A", "E"]) # With single-level Index s = df.set_index("A")["B"] result = s.reset_index(level=levels[0]) tm.assert_frame_equal(result, df[["A", "B"]]) result = s.reset_index(level=levels[:1]) tm.assert_frame_equal(result, df[["A", "B"]]) result = s.reset_index(level=levels[0], drop=True) tm.assert_series_equal(result, df["B"]) with pytest.raises(IndexError, match="Too many levels"): s.reset_index(level=[0, 1, 2]) # Check that .reset_index([],drop=True) doesn't fail result = Series(range(4)).reset_index([], drop=True) expected = Series(range(4)) tm.assert_series_equal(result, expected) def test_reset_index_range(self): # GH 12071 s = Series(range(2), name="A", dtype="int64") series_result = s.reset_index() assert isinstance(series_result.index, RangeIndex) series_expected = DataFrame( [[0, 0], [1, 1]], columns=["index", "A"], index=RangeIndex(stop=2) ) tm.assert_frame_equal(series_result, series_expected) def test_reset_index_drop_errors(self): # GH 20925 # KeyError raised for series index when passed level name is missing s = Series(range(4)) with pytest.raises(KeyError, match="does not match index name"): s.reset_index("wrong", drop=True) with pytest.raises(KeyError, match="does not match index name"): s.reset_index("wrong") # KeyError raised for series when level to be dropped is missing s = Series(range(4), index=MultiIndex.from_product([[1, 2]] * 2)) with pytest.raises(KeyError, match="not found"): s.reset_index("wrong", drop=True) @pytest.mark.parametrize( "array, dtype", [ (["a", "b"], object), ( pd.period_range("12-1-2000", periods=2, freq="Q-DEC"), pd.PeriodDtype(freq="Q-DEC"), ), ], ) def test_reset_index_dtypes_on_empty_series_with_multiindex(array, dtype): # GH 19602 - Preserve dtype on empty Series with MultiIndex idx = MultiIndex.from_product([[0, 1], [0.5, 1.0], array]) result = Series(dtype=object, index=idx)[:0].reset_index().dtypes expected = Series( {"level_0": np.int64, "level_1": np.float64, "level_2": dtype, 0: object} ) tm.assert_series_equal(result, expected)