Na}{

File Content: `test_pop.py`

import numpy as np

from pandas import (
    DataFrame,
    MultiIndex,
    Series,
)
import pandas._testing as tm


class TestDataFramePop:
    def test_pop(self, float_frame):
        float_frame.columns.name = "baz"

        float_frame.pop("A")
        assert "A" not in float_frame

        float_frame["foo"] = "bar"
        float_frame.pop("foo")
        assert "foo" not in float_frame
        assert float_frame.columns.name == "baz"

        # gh-10912: inplace ops cause caching issue
        a = DataFrame([[1, 2, 3], [4, 5, 6]], columns=["A", "B", "C"], index=["X", "Y"])
        b = a.pop("B")
        b += 1

        # original frame
        expected = DataFrame([[1, 3], [4, 6]], columns=["A", "C"], index=["X", "Y"])
        tm.assert_frame_equal(a, expected)

        # result
        expected = Series([2, 5], index=["X", "Y"], name="B") + 1
        tm.assert_series_equal(b, expected)

    def test_pop_non_unique_cols(self):
        df = DataFrame({0: [0, 1], 1: [0, 1], 2: [4, 5]})
        df.columns = ["a", "b", "a"]

        res = df.pop("a")
        assert type(res) == DataFrame
        assert len(res) == 2
        assert len(df.columns) == 1
        assert "b" in df.columns
        assert "a" not in df.columns
        assert len(df.index) == 2

    def test_mixed_depth_pop(self):
        arrays = [
            ["a", "top", "top", "routine1", "routine1", "routine2"],
            ["", "OD", "OD", "result1", "result2", "result1"],
            ["", "wx", "wy", "", "", ""],
        ]

        tuples = sorted(zip(*arrays))
        index = MultiIndex.from_tuples(tuples)
        df = DataFrame(np.random.randn(4, 6), columns=index)

        df1 = df.copy()
        df2 = df.copy()
        result = df1.pop("a")
        expected = df2.pop(("a", "", ""))
        tm.assert_series_equal(expected, result, check_names=False)
        tm.assert_frame_equal(df1, df2)
        assert result.name == "a"

        expected = df1["top"]
        df1 = df1.drop(["top"], axis=1)
        result = df2.pop("top")
        tm.assert_frame_equal(expected, result)
        tm.assert_frame_equal(df1, df2)

Name	Size	Permission
__pycache__	---	2755
__init__.py	229 bytes	0644
test_add_prefix_suffix.py	1910 bytes	0644
test_align.py	15901 bytes	0644
test_asfreq.py	7579 bytes	0644
test_asof.py	6613 bytes	0644
test_assign.py	2982 bytes	0644
test_astype.py	31448 bytes	0644
test_at_time.py	4462 bytes	0644
test_between_time.py	7734 bytes	0644
test_clip.py	6261 bytes	0644
test_combine.py	1359 bytes	0644
test_combine_first.py	18871 bytes	0644
test_compare.py	9464 bytes	0644
test_convert_dtypes.py	6707 bytes	0644
test_copy.py	1828 bytes	0644
test_count.py	1083 bytes	0644
test_cov_corr.py	16643 bytes	0644
test_describe.py	14484 bytes	0644
test_diff.py	9849 bytes	0644
test_dot.py	3899 bytes	0644
test_drop.py	19801 bytes	0644
test_drop_duplicates.py	14512 bytes	0644
test_droplevel.py	1253 bytes	0644
test_dropna.py	10163 bytes	0644
test_dtypes.py	4831 bytes	0644
test_duplicated.py	3221 bytes	0644
test_equals.py	2893 bytes	0644
test_explode.py	8810 bytes	0644
test_fillna.py	26574 bytes	0644
test_filter.py	4930 bytes	0644
test_first_and_last.py	3127 bytes	0644
test_first_valid_index.py	2520 bytes	0644
test_get_numeric_data.py	3216 bytes	0644
test_head_tail.py	1911 bytes	0644
test_infer_objects.py	1241 bytes	0644
test_interpolate.py	14772 bytes	0644
test_is_homogeneous_dtype.py	1422 bytes	0644
test_isetitem.py	965 bytes	0644
test_isin.py	7323 bytes	0644
test_join.py	17273 bytes	0644
test_matmul.py	2847 bytes	0644
test_nlargest.py	8160 bytes	0644
test_pct_change.py	4526 bytes	0644
test_pipe.py	1023 bytes	0644
test_pop.py	2116 bytes	0644
test_quantile.py	36555 bytes	0644
test_rank.py	16804 bytes	0644
test_reindex.py	45599 bytes	0644
test_reindex_like.py	1187 bytes	0644
test_rename.py	15326 bytes	0644
test_rename_axis.py	4091 bytes	0644
test_reorder_levels.py	2729 bytes	0644
test_replace.py	58188 bytes	0644
test_reset_index.py	27809 bytes	0644
test_round.py	7944 bytes	0644
test_sample.py	13182 bytes	0644
test_select_dtypes.py	16528 bytes	0644
test_set_axis.py	4652 bytes	0644
test_set_index.py	25153 bytes	0644
test_shift.py	22617 bytes	0644
test_sort_index.py	31041 bytes	0644
test_sort_values.py	32769 bytes	0644
test_swapaxes.py	885 bytes	0644
test_swaplevel.py	1277 bytes	0644
test_to_csv.py	48009 bytes	0644
test_to_dict.py	17266 bytes	0644
test_to_dict_of_blocks.py	3020 bytes	0644
test_to_numpy.py	1442 bytes	0644
test_to_period.py	2708 bytes	0644
test_to_records.py	18414 bytes	0644
test_to_timestamp.py	5754 bytes	0644
test_transpose.py	4488 bytes	0644
test_truncate.py	5006 bytes	0644
test_tz_convert.py	4722 bytes	0644
test_tz_localize.py	2084 bytes	0644
test_update.py	5836 bytes	0644
test_value_counts.py	4749 bytes	0644
test_values.py	9349 bytes	0644

File Content: test_pop.py

File Content: `test_pop.py`