Add accessors to class attribute (pandas-dev#20780)

Re-using the _accessors attribute, since nothing seemed to rely on it
being a frozenset.

Closes pandas-dev#19963

Author: TomAugspurger

pandas/core/accessor.py

@@ -148,6 +148,7 @@ def decorator(accessor):
                 UserWarning,
                 stacklevel=2)
         setattr(cls, name, CachedAccessor(name, accessor))
+        cls._accessors.add(name)
         return accessor
     return decorator
 

pandas/core/frame.py

@@ -326,6 +326,7 @@ def _constructor(self):
     _constructor_sliced = Series
     _deprecations = NDFrame._deprecations | frozenset(
         ['sortlevel', 'get_value', 'set_value', 'from_csv', 'from_items'])
+    _accessors = set()
 
     @property
     def _constructor_expanddim(self):

pandas/core/indexes/base.py

@@ -242,7 +242,8 @@ class Index(IndexOpsMixin, PandasObject):
 
     _engine_type = libindex.ObjectEngine
 
-    _accessors = frozenset(['str'])
+    _accessors = set(['str'])
+
     str = CachedAccessor("str", StringMethods)
 
     def __new__(cls, data=None, dtype=None, copy=False, name=None,

pandas/core/series.py

@@ -153,7 +153,7 @@ class Series(base.IndexOpsMixin, generic.NDFrame):
         Copy input data
     """
     _metadata = ['name']
-    _accessors = frozenset(['dt', 'cat', 'str'])
+    _accessors = set(['dt', 'cat', 'str'])
     _deprecations = generic.NDFrame._deprecations | frozenset(
         ['asobject', 'sortlevel', 'reshape', 'get_value', 'set_value',
          'from_csv', 'valid'])

pandas/tests/test_register_accessor.py

@@ -17,6 +17,7 @@ def ensure_removed(obj, attr):
             delattr(obj, attr)
         except AttributeError:
             pass
+        obj._accessors.discard(attr)
 
 
 class MyAccessor(object):
@@ -38,13 +39,14 @@ def method(self):
     (pd.DataFrame, pd.api.extensions.register_dataframe_accessor),
     (pd.Index, pd.api.extensions.register_index_accessor)
 ])
-def test_series_register(obj, registrar):
+def test_register(obj, registrar):
     with ensure_removed(obj, 'mine'):
         before = set(dir(obj))
         registrar('mine')(MyAccessor)
         assert obj([]).mine.prop == 'item'
         after = set(dir(obj))
         assert (before ^ after) == {'mine'}
+        assert 'mine' in obj._accessors
 
 
 def test_accessor_works():

scripts/validate_docstrings.py

@@ -182,7 +182,8 @@ def doc_parameters(self):
     @property
     def signature_parameters(self):
         if (inspect.isclass(self.method_obj)
-                and self.method_name.split('.')[-1] in {'dt', 'str', 'cat'}):
+                and self.method_name.split('.')[-1] in
+                self.method_obj._accessors):
             # accessor classes have a signature, but don't want to show this
             return tuple()
         try: