Beta Shell v2.0

File Manager

~ /usr/local/python-3.13/lib/python3.13/site-packages/pandas/tests/frame/methods

Name	Size	Perms	Modified	Actions
[ .. / .. ]
__init__.py	229 B	-rw-r--r--	2025-10-22 11:54:19	Edit Del
__pycache__	-	drwxr-xr-x	2025-10-22 11:54:24	Del
test_add_prefix_suffix.py	1.87 KB	-rw-r--r--	2025-10-22 11:54:19	Edit Del
test_align.py	17.52 KB	-rw-r--r--	2025-10-22 11:54:19	Edit Del
test_asfreq.py	9.12 KB	-rw-r--r--	2025-10-22 11:54:19	Edit Del
test_asof.py	6.57 KB	-rw-r--r--	2025-10-22 11:54:19	Edit Del
test_assign.py	2.91 KB	-rw-r--r--	2025-10-22 11:54:19	Edit Del
test_astype.py	31.94 KB	-rw-r--r--	2025-10-22 11:54:19	Edit Del
test_at_time.py	4.6 KB	-rw-r--r--	2025-10-22 11:54:19	Edit Del
test_between_time.py	7.89 KB	-rw-r--r--	2025-10-22 11:54:19	Edit Del
test_clip.py	7.38 KB	-rw-r--r--	2025-10-22 11:54:19	Edit Del
test_combine.py	1.33 KB	-rw-r--r--	2025-10-22 11:54:19	Edit Del
test_combine_first.py	19.26 KB	-rw-r--r--	2025-10-22 11:54:19	Edit Del
test_compare.py	9.39 KB	-rw-r--r--	2025-10-22 11:54:19	Edit Del
test_convert_dtypes.py	7.66 KB	-rw-r--r--	2025-10-22 11:54:19	Edit Del
test_copy.py	1.83 KB	-rw-r--r--	2025-10-22 11:54:19	Edit Del
test_count.py	1.06 KB	-rw-r--r--	2025-10-22 11:54:19	Edit Del
test_cov_corr.py	17.46 KB	-rw-r--r--	2025-10-22 11:54:19	Edit Del
test_describe.py	14.16 KB	-rw-r--r--	2025-10-22 11:54:19	Edit Del
test_diff.py	9.86 KB	-rw-r--r--	2025-10-22 11:54:19	Edit Del
test_dot.py	4.51 KB	-rw-r--r--	2025-10-22 11:54:19	Edit Del
test_drop.py	19.88 KB	-rw-r--r--	2025-10-22 11:54:19	Edit Del
test_drop_duplicates.py	14.16 KB	-rw-r--r--	2025-10-22 11:54:19	Edit Del
test_droplevel.py	1.22 KB	-rw-r--r--	2025-10-22 11:54:19	Edit Del
test_dropna.py	10.07 KB	-rw-r--r--	2025-10-22 11:54:19	Edit Del
test_dtypes.py	4.88 KB	-rw-r--r--	2025-10-22 11:54:19	Edit Del
test_duplicated.py	3.23 KB	-rw-r--r--	2025-10-22 11:54:19	Edit Del
test_equals.py	2.93 KB	-rw-r--r--	2025-10-22 11:54:19	Edit Del
test_explode.py	8.85 KB	-rw-r--r--	2025-10-22 11:54:19	Edit Del
test_fillna.py	32.63 KB	-rw-r--r--	2025-10-22 11:54:19	Edit Del
test_filter.py	5.29 KB	-rw-r--r--	2025-10-22 11:54:19	Edit Del
test_first_and_last.py	5.22 KB	-rw-r--r--	2025-10-22 11:54:19	Edit Del
test_first_valid_index.py	2.51 KB	-rw-r--r--	2025-10-22 11:54:19	Edit Del
test_get_numeric_data.py	3.29 KB	-rw-r--r--	2025-10-22 11:54:19	Edit Del
test_head_tail.py	1.89 KB	-rw-r--r--	2025-10-22 11:54:19	Edit Del
test_infer_objects.py	1.21 KB	-rw-r--r--	2025-10-22 11:54:19	Edit Del
test_info.py	17.5 KB	-rw-r--r--	2025-10-22 11:54:19	Edit Del
test_interpolate.py	19.93 KB	-rw-r--r--	2025-10-22 11:54:19	Edit Del
test_is_homogeneous_dtype.py	1.42 KB	-rw-r--r--	2025-10-22 11:54:19	Edit Del
test_isetitem.py	1.39 KB	-rw-r--r--	2025-10-22 11:54:19	Edit Del
test_isin.py	7.42 KB	-rw-r--r--	2025-10-22 11:54:19	Edit Del
test_iterrows.py	338 B	-rw-r--r--	2025-10-22 11:54:19	Edit Del
test_join.py	17.11 KB	-rw-r--r--	2025-10-22 11:54:19	Edit Del
test_map.py	5.85 KB	-rw-r--r--	2025-10-22 11:54:19	Edit Del
test_matmul.py	3.06 KB	-rw-r--r--	2025-10-22 11:54:19	Edit Del
test_nlargest.py	8 KB	-rw-r--r--	2025-10-22 11:54:19	Edit Del
test_pct_change.py	6.38 KB	-rw-r--r--	2025-10-22 11:54:19	Edit Del
test_pipe.py	1023 B	-rw-r--r--	2025-10-22 11:54:19	Edit Del
test_pop.py	2.17 KB	-rw-r--r--	2025-10-22 11:54:19	Edit Del
test_quantile.py	35.73 KB	-rw-r--r--	2025-10-22 11:54:19	Edit Del
test_rank.py	17.14 KB	-rw-r--r--	2025-10-22 11:54:19	Edit Del
test_reindex.py	47.21 KB	-rw-r--r--	2025-10-22 11:54:19	Edit Del
test_reindex_like.py	1.16 KB	-rw-r--r--	2025-10-22 11:54:19	Edit Del
test_rename.py	14.99 KB	-rw-r--r--	2025-10-22 11:54:19	Edit Del
test_rename_axis.py	4 KB	-rw-r--r--	2025-10-22 11:54:19	Edit Del
test_reorder_levels.py	2.67 KB	-rw-r--r--	2025-10-22 11:54:19	Edit Del
test_replace.py	61.28 KB	-rw-r--r--	2025-10-22 11:54:19	Edit Del
test_reset_index.py	28.47 KB	-rw-r--r--	2025-10-22 11:54:19	Edit Del
test_round.py	7.79 KB	-rw-r--r--	2025-10-22 11:54:19	Edit Del
test_sample.py	13.12 KB	-rw-r--r--	2025-10-22 11:54:19	Edit Del
test_select_dtypes.py	17.75 KB	-rw-r--r--	2025-10-22 11:54:19	Edit Del
test_set_axis.py	4.5 KB	-rw-r--r--	2025-10-22 11:54:19	Edit Del
test_set_index.py	25.95 KB	-rw-r--r--	2025-10-22 11:54:19	Edit Del
test_shift.py	27.08 KB	-rw-r--r--	2025-10-22 11:54:19	Edit Del
test_size.py	571 B	-rw-r--r--	2025-10-22 11:54:19	Edit Del
test_sort_index.py	34.01 KB	-rw-r--r--	2025-10-22 11:54:19	Edit Del
test_sort_values.py	32.21 KB	-rw-r--r--	2025-10-22 11:54:19	Edit Del
test_swapaxes.py	1.43 KB	-rw-r--r--	2025-10-22 11:54:19	Edit Del
test_swaplevel.py	1.25 KB	-rw-r--r--	2025-10-22 11:54:19	Edit Del
test_to_csv.py	50.35 KB	-rw-r--r--	2025-10-22 11:54:19	Edit Del
test_to_dict.py	18.2 KB	-rw-r--r--	2025-10-22 11:54:19	Edit Del
test_to_dict_of_blocks.py	2.58 KB	-rw-r--r--	2025-10-22 11:54:19	Edit Del
test_to_numpy.py	1.87 KB	-rw-r--r--	2025-10-22 11:54:19	Edit Del
test_to_period.py	2.8 KB	-rw-r--r--	2025-10-22 11:54:19	Edit Del
test_to_records.py	18.12 KB	-rw-r--r--	2025-10-22 11:54:19	Edit Del
test_to_timestamp.py	5.73 KB	-rw-r--r--	2025-10-22 11:54:19	Edit Del
test_transpose.py	6.67 KB	-rw-r--r--	2025-10-22 11:54:19	Edit Del
test_truncate.py	5.09 KB	-rw-r--r--	2025-10-22 11:54:19	Edit Del
test_tz_convert.py	4.6 KB	-rw-r--r--	2025-10-22 11:54:19	Edit Del
test_tz_localize.py	2.04 KB	-rw-r--r--	2025-10-22 11:54:19	Edit Del
test_update.py	6.73 KB	-rw-r--r--	2025-10-22 11:54:19	Edit Del
test_value_counts.py	5.43 KB	-rw-r--r--	2025-10-22 11:54:19	Edit Del
test_values.py	9.19 KB	-rw-r--r--	2025-10-22 11:54:19	Edit Del

Editing: test_clip.py (7.38 KB)

import numpy as np
import pytest

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


class TestDataFrameClip:
    def test_clip(self, float_frame):
        median = float_frame.median().median()
        original = float_frame.copy()

        double = float_frame.clip(upper=median, lower=median)
        assert not (double.values != median).any()

        # Verify that float_frame was not changed inplace
        assert (float_frame.values == original.values).all()

    def test_inplace_clip(self, float_frame):
        # GH#15388
        median = float_frame.median().median()
        frame_copy = float_frame.copy()

        return_value = frame_copy.clip(upper=median, lower=median, inplace=True)
        assert return_value is None
        assert not (frame_copy.values != median).any()

    def test_dataframe_clip(self):
        # GH#2747
        df = DataFrame(np.random.default_rng(2).standard_normal((1000, 2)))

        for lb, ub in [(-1, 1), (1, -1)]:
            clipped_df = df.clip(lb, ub)

            lb, ub = min(lb, ub), max(ub, lb)
            lb_mask = df.values <= lb
            ub_mask = df.values >= ub
            mask = ~lb_mask & ~ub_mask
            assert (clipped_df.values[lb_mask] == lb).all()
            assert (clipped_df.values[ub_mask] == ub).all()
            assert (clipped_df.values[mask] == df.values[mask]).all()

    def test_clip_mixed_numeric(self):
        # clip on mixed integer or floats
        # GH#24162, clipping now preserves numeric types per column
        df = DataFrame({"A": [1, 2, 3], "B": [1.0, np.nan, 3.0]})
        result = df.clip(1, 2)
        expected = DataFrame({"A": [1, 2, 2], "B": [1.0, np.nan, 2.0]})
        tm.assert_frame_equal(result, expected)

        df = DataFrame([[1, 2, 3.4], [3, 4, 5.6]], columns=["foo", "bar", "baz"])
        expected = df.dtypes
        result = df.clip(upper=3).dtypes
        tm.assert_series_equal(result, expected)

    @pytest.mark.parametrize("inplace", [True, False])
    def test_clip_against_series(self, inplace):
        # GH#6966

        df = DataFrame(np.random.default_rng(2).standard_normal((1000, 2)))
        lb = Series(np.random.default_rng(2).standard_normal(1000))
        ub = lb + 1

        original = df.copy()
        clipped_df = df.clip(lb, ub, axis=0, inplace=inplace)

        if inplace:
            clipped_df = df

        for i in range(2):
            lb_mask = original.iloc[:, i] <= lb
            ub_mask = original.iloc[:, i] >= ub
            mask = ~lb_mask & ~ub_mask

            result = clipped_df.loc[lb_mask, i]
            tm.assert_series_equal(result, lb[lb_mask], check_names=False)
            assert result.name == i

            result = clipped_df.loc[ub_mask, i]
            tm.assert_series_equal(result, ub[ub_mask], check_names=False)
            assert result.name == i

            tm.assert_series_equal(clipped_df.loc[mask, i], df.loc[mask, i])

    @pytest.mark.parametrize("inplace", [True, False])
    @pytest.mark.parametrize("lower", [[2, 3, 4], np.asarray([2, 3, 4])])
    @pytest.mark.parametrize(
        "axis,res",
        [
            (0, [[2.0, 2.0, 3.0], [4.0, 5.0, 6.0], [7.0, 7.0, 7.0]]),
            (1, [[2.0, 3.0, 4.0], [4.0, 5.0, 6.0], [5.0, 6.0, 7.0]]),
        ],
    )
    def test_clip_against_list_like(self, inplace, lower, axis, res):
        # GH#15390
        arr = np.array([[1.0, 2.0, 3.0], [4.0, 5.0, 6.0], [7.0, 8.0, 9.0]])

        original = DataFrame(
            arr, columns=["one", "two", "three"], index=["a", "b", "c"]
        )

        result = original.clip(lower=lower, upper=[5, 6, 7], axis=axis, inplace=inplace)

        expected = DataFrame(res, columns=original.columns, index=original.index)
        if inplace:
            result = original
        tm.assert_frame_equal(result, expected, check_exact=True)

    @pytest.mark.parametrize("axis", [0, 1, None])
    def test_clip_against_frame(self, axis):
        df = DataFrame(np.random.default_rng(2).standard_normal((1000, 2)))
        lb = DataFrame(np.random.default_rng(2).standard_normal((1000, 2)))
        ub = lb + 1

        clipped_df = df.clip(lb, ub, axis=axis)

        lb_mask = df <= lb
        ub_mask = df >= ub
        mask = ~lb_mask & ~ub_mask

        tm.assert_frame_equal(clipped_df[lb_mask], lb[lb_mask])
        tm.assert_frame_equal(clipped_df[ub_mask], ub[ub_mask])
        tm.assert_frame_equal(clipped_df[mask], df[mask])

    def test_clip_against_unordered_columns(self):
        # GH#20911
        df1 = DataFrame(
            np.random.default_rng(2).standard_normal((1000, 4)),
            columns=["A", "B", "C", "D"],
        )
        df2 = DataFrame(
            np.random.default_rng(2).standard_normal((1000, 4)),
            columns=["D", "A", "B", "C"],
        )
        df3 = DataFrame(df2.values - 1, columns=["B", "D", "C", "A"])
        result_upper = df1.clip(lower=0, upper=df2)
        expected_upper = df1.clip(lower=0, upper=df2[df1.columns])
        result_lower = df1.clip(lower=df3, upper=3)
        expected_lower = df1.clip(lower=df3[df1.columns], upper=3)
        result_lower_upper = df1.clip(lower=df3, upper=df2)
        expected_lower_upper = df1.clip(lower=df3[df1.columns], upper=df2[df1.columns])
        tm.assert_frame_equal(result_upper, expected_upper)
        tm.assert_frame_equal(result_lower, expected_lower)
        tm.assert_frame_equal(result_lower_upper, expected_lower_upper)

    def test_clip_with_na_args(self, float_frame):
        """Should process np.nan argument as None"""
        # GH#17276
        tm.assert_frame_equal(float_frame.clip(np.nan), float_frame)
        tm.assert_frame_equal(float_frame.clip(upper=np.nan, lower=np.nan), float_frame)

        # GH#19992 and adjusted in GH#40420
        df = DataFrame({"col_0": [1, 2, 3], "col_1": [4, 5, 6], "col_2": [7, 8, 9]})

        msg = "Downcasting behavior in Series and DataFrame methods 'where'"
        # TODO: avoid this warning here?  seems like we should never be upcasting
        #  in the first place?
        with tm.assert_produces_warning(FutureWarning, match=msg):
            result = df.clip(lower=[4, 5, np.nan], axis=0)
        expected = DataFrame(
            {"col_0": [4, 5, 3], "col_1": [4, 5, 6], "col_2": [7, 8, 9]}
        )
        tm.assert_frame_equal(result, expected)

        result = df.clip(lower=[4, 5, np.nan], axis=1)
        expected = DataFrame(
            {"col_0": [4, 4, 4], "col_1": [5, 5, 6], "col_2": [7, 8, 9]}
        )
        tm.assert_frame_equal(result, expected)

        # GH#40420
        data = {"col_0": [9, -3, 0, -1, 5], "col_1": [-2, -7, 6, 8, -5]}
        df = DataFrame(data)
        t = Series([2, -4, np.nan, 6, 3])
        with tm.assert_produces_warning(FutureWarning, match=msg):
            result = df.clip(lower=t, axis=0)
        expected = DataFrame({"col_0": [9, -3, 0, 6, 5], "col_1": [2, -4, 6, 8, 3]})
        tm.assert_frame_equal(result, expected)

    def test_clip_int_data_with_float_bound(self):
        # GH51472
        df = DataFrame({"a": [1, 2, 3]})
        result = df.clip(lower=1.5)
        expected = DataFrame({"a": [1.5, 2.0, 3.0]})
        tm.assert_frame_equal(result, expected)

    def test_clip_with_list_bound(self):
        # GH#54817
        df = DataFrame([1, 5])
        expected = DataFrame([3, 5])
        result = df.clip([3])
        tm.assert_frame_equal(result, expected)

        expected = DataFrame([1, 3])
        result = df.clip(upper=[3])
        tm.assert_frame_equal(result, expected)