Beta Shell v2.0

File Manager

~ /usr/local/python-3.11/lib/python3.11/site-packages/pandas/tests/indexes

Name	Size	Perms	Modified	Actions
[ .. / .. ]
__init__.py	0 B	-rw-r--r--	2024-04-12 10:59:24	Edit Del
__pycache__	-	drwxr-xr-x	2024-04-12 10:59:30	Del
base_class	-	drwxr-xr-x	2024-04-12 10:59:29	Del
categorical	-	drwxr-xr-x	2024-04-12 10:59:29	Del
conftest.py	987 B	-rw-r--r--	2024-04-12 10:59:24	Edit Del
datetimelike_	-	drwxr-xr-x	2024-04-12 10:59:29	Del
datetimes	-	drwxr-xr-x	2024-04-12 10:59:29	Del
interval	-	drwxr-xr-x	2024-04-12 10:59:29	Del
multi	-	drwxr-xr-x	2024-04-12 10:59:29	Del
numeric	-	drwxr-xr-x	2024-04-12 10:59:29	Del
object	-	drwxr-xr-x	2024-04-12 10:59:29	Del
period	-	drwxr-xr-x	2024-04-12 10:59:29	Del
ranges	-	drwxr-xr-x	2024-04-12 10:59:29	Del
test_any_index.py	5.01 KB	-rw-r--r--	2024-04-12 10:59:24	Edit Del
test_base.py	59.23 KB	-rw-r--r--	2024-04-12 10:59:24	Edit Del
test_common.py	17.45 KB	-rw-r--r--	2024-04-12 10:59:24	Edit Del
test_datetimelike.py	5.47 KB	-rw-r--r--	2024-04-12 10:59:24	Edit Del
test_engines.py	6.54 KB	-rw-r--r--	2024-04-12 10:59:24	Edit Del
test_frozen.py	3.05 KB	-rw-r--r--	2024-04-12 10:59:24	Edit Del
test_index_new.py	14.57 KB	-rw-r--r--	2024-04-12 10:59:24	Edit Del
test_indexing.py	11.04 KB	-rw-r--r--	2024-04-12 10:59:24	Edit Del
test_numpy_compat.py	5.64 KB	-rw-r--r--	2024-04-12 10:59:24	Edit Del
test_old_base.py	38.99 KB	-rw-r--r--	2024-04-12 10:59:24	Edit Del
test_setops.py	32.17 KB	-rw-r--r--	2024-04-12 10:59:24	Edit Del
test_subclass.py	1.03 KB	-rw-r--r--	2024-04-12 10:59:24	Edit Del
timedeltas	-	drwxr-xr-x	2024-04-12 10:59:30	Del

Editing: test_frozen.py (3.05 KB)

import re

import pytest

from pandas.core.indexes.frozen import FrozenList


@pytest.fixture
def lst():
    return [1, 2, 3, 4, 5]


@pytest.fixture
def container(lst):
    return FrozenList(lst)


@pytest.fixture
def unicode_container():
    return FrozenList(["\u05d0", "\u05d1", "c"])


class TestFrozenList:
    def check_mutable_error(self, *args, **kwargs):
        # Pass whatever function you normally would to pytest.raises
        # (after the Exception kind).
        mutable_regex = re.compile("does not support mutable operations")
        msg = "'(_s)?re.(SRE_)?Pattern' object is not callable"
        with pytest.raises(TypeError, match=msg):
            mutable_regex(*args, **kwargs)

    def test_no_mutable_funcs(self, container):
        def setitem():
            container[0] = 5

        self.check_mutable_error(setitem)

        def setslice():
            container[1:2] = 3

        self.check_mutable_error(setslice)

        def delitem():
            del container[0]

        self.check_mutable_error(delitem)

        def delslice():
            del container[0:3]

        self.check_mutable_error(delslice)

        mutable_methods = ("extend", "pop", "remove", "insert")

        for meth in mutable_methods:
            self.check_mutable_error(getattr(container, meth))

    def test_slicing_maintains_type(self, container, lst):
        result = container[1:2]
        expected = lst[1:2]
        self.check_result(result, expected)

    def check_result(self, result, expected):
        assert isinstance(result, FrozenList)
        assert result == expected

    def test_string_methods_dont_fail(self, container):
        repr(container)
        str(container)
        bytes(container)

    def test_tricky_container(self, unicode_container):
        repr(unicode_container)
        str(unicode_container)

    def test_add(self, container, lst):
        result = container + (1, 2, 3)
        expected = FrozenList(lst + [1, 2, 3])
        self.check_result(result, expected)

        result = (1, 2, 3) + container
        expected = FrozenList([1, 2, 3] + lst)
        self.check_result(result, expected)

    def test_iadd(self, container, lst):
        q = r = container

        q += [5]
        self.check_result(q, lst + [5])

        # Other shouldn't be mutated.
        self.check_result(r, lst)

    def test_union(self, container, lst):
        result = container.union((1, 2, 3))
        expected = FrozenList(lst + [1, 2, 3])
        self.check_result(result, expected)

    def test_difference(self, container):
        result = container.difference([2])
        expected = FrozenList([1, 3, 4, 5])
        self.check_result(result, expected)

    def test_difference_dupe(self):
        result = FrozenList([1, 2, 3, 2]).difference([2])
        expected = FrozenList([1, 3])
        self.check_result(result, expected)

    def test_tricky_container_to_bytes_raises(self, unicode_container):
        # GH 26447
        msg = "^'str' object cannot be interpreted as an integer$"
        with pytest.raises(TypeError, match=msg):
            bytes(unicode_container)