FM / tests

/usr/local/lib64/python3.6/site-packages/pandas/tests/

/usr/local/lib64/python3.6/site-packages/pandas/tests

Name	Size	Mode	Actions
api/	-	0755	rm
arithmetic/	-	0755	rm
arrays/	-	0755	rm
base/	-	0755	rm
computation/	-	0755	rm
config/	-	0755	rm
dtypes/	-	0755	rm
extension/	-	0755	rm
frame/	-	0755	rm
generic/	-	0755	rm
groupby/	-	0755	rm
indexes/	-	0755	rm
indexing/	-	0755	rm
internals/	-	0755	rm
io/	-	0755	rm
plotting/	-	0755	rm
reductions/	-	0755	rm
resample/	-	0755	rm
reshape/	-	0755	rm
scalar/	-	0755	rm
series/	-	0755	rm
tools/	-	0755	rm
tseries/	-	0755	rm
tslibs/	-	0755	rm
util/	-	0755	rm
window/	-	0755	rm
__pycache__/	-	0755	rm
test_aggregation.py	2772	0644	edit dl rm
test_algos.py	82574	0644	edit dl rm
test_common.py	4447	0644	edit dl rm
test_downstream.py	5163	0644	edit dl rm
test_errors.py	1678	0644	edit dl rm
test_expressions.py	14023	0644	edit dl rm
test_join.py	9296	0644	edit dl rm
test_lib.py	7958	0644	edit dl rm
test_multilevel.py	65726	0644	edit dl rm
test_nanops.py	37847	0644	edit dl rm
test_optional_dependency.py	1518	0644	edit dl rm
test_register_accessor.py	2351	0644	edit dl rm
test_sorting.py	17786	0644	edit dl rm
test_strings.py	133766	0644	edit dl rm
test_take.py	16875	0644	edit dl rm
__init__.py	0	0644	edit dl rm

Edit: /usr/local/lib64/python3.6/site-packages/pandas/tests/test_register_accessor.py (2351B)

import contextlib

import pytest

import pandas as pd
import pandas._testing as tm

@contextlib.contextmanager
def ensure_removed(obj, attr):
    """Ensure that an attribute added to 'obj' during the test is
    removed when we're done
    """
    try:
        yield
    finally:
        try:
            delattr(obj, attr)
        except AttributeError:
            pass
        obj._accessors.discard(attr)

class MyAccessor:
    def __init__(self, obj):
        self.obj = obj
        self.item = "item"

@property
    def prop(self):
        return self.item

def method(self):
        return self.item

@pytest.mark.parametrize(
    "obj, registrar",
    [
        (pd.Series, pd.api.extensions.register_series_accessor),
        (pd.DataFrame, pd.api.extensions.register_dataframe_accessor),
        (pd.Index, pd.api.extensions.register_index_accessor),
    ],
)
def test_register(obj, registrar):
    with ensure_removed(obj, "mine"):
        before = set(dir(obj))
        registrar("mine")(MyAccessor)
        o = obj([]) if obj is not pd.Series else obj([], dtype=object)
        assert o.mine.prop == "item"
        after = set(dir(obj))
        assert (before ^ after) == {"mine"}
        assert "mine" in obj._accessors

def test_accessor_works():
    with ensure_removed(pd.Series, "mine"):
        pd.api.extensions.register_series_accessor("mine")(MyAccessor)

s = pd.Series([1, 2])
        assert s.mine.obj is s

assert s.mine.prop == "item"
        assert s.mine.method() == "item"

def test_overwrite_warns():
    # Need to restore mean
    mean = pd.Series.mean
    try:
        with tm.assert_produces_warning(UserWarning) as w:
            pd.api.extensions.register_series_accessor("mean")(MyAccessor)
            s = pd.Series([1, 2])
            assert s.mean.prop == "item"
        msg = str(w[0].message)
        assert "mean" in msg
        assert "MyAccessor" in msg
        assert "Series" in msg
    finally:
        pd.Series.mean = mean

def test_raises_attribute_error():

with ensure_removed(pd.Series, "bad"):

@pd.api.extensions.register_series_accessor("bad")
        class Bad:
            def __init__(self, data):
                raise AttributeError("whoops")

with pytest.raises(AttributeError, match="whoops"):
            pd.Series([], dtype=object).bad

import contextlib import pytest import pandas as pd import pandas._testing as tm @contextlib.contextmanager def ensure_removed(obj, attr): """Ensure that an attribute added to 'obj' during the test is removed when we're done """ try: yield finally: try: delattr(obj, attr) except AttributeError: pass obj._accessors.discard(attr) class MyAccessor: def __init__(self, obj): self.obj = obj self.item = "item" @property def prop(self): return self.item def method(self): return self.item @pytest.mark.parametrize( "obj, registrar", [ (pd.Series, pd.api.extensions.register_series_accessor), (pd.DataFrame, pd.api.extensions.register_dataframe_accessor), (pd.Index, pd.api.extensions.register_index_accessor), ], ) def test_register(obj, registrar): with ensure_removed(obj, "mine"): before = set(dir(obj)) registrar("mine")(MyAccessor) o = obj([]) if obj is not pd.Series else obj([], dtype=object) assert o.mine.prop == "item" after = set(dir(obj)) assert (before ^ after) == {"mine"} assert "mine" in obj._accessors def test_accessor_works(): with ensure_removed(pd.Series, "mine"): pd.api.extensions.register_series_accessor("mine")(MyAccessor) s = pd.Series([1, 2]) assert s.mine.obj is s assert s.mine.prop == "item" assert s.mine.method() == "item" def test_overwrite_warns(): # Need to restore mean mean = pd.Series.mean try: with tm.assert_produces_warning(UserWarning) as w: pd.api.extensions.register_series_accessor("mean")(MyAccessor) s = pd.Series([1, 2]) assert s.mean.prop == "item" msg = str(w[0].message) assert "mean" in msg assert "MyAccessor" in msg assert "Series" in msg finally: pd.Series.mean = mean def test_raises_attribute_error(): with ensure_removed(pd.Series, "bad"): @pd.api.extensions.register_series_accessor("bad") class Bad: def __init__(self, data): raise AttributeError("whoops") with pytest.raises(AttributeError, match="whoops"): pd.Series([], dtype=object).bad