Blacken

Author: jseabold

pandas_select/__init__.py

@@ -4,37 +4,43 @@
 
 import pandas as pd
 from pandas.core.accessor import CachedAccessor
-from pandas.core.indexes.accessors import (CombinedDatetimelikeProperties,
-                                           DatetimeProperties,
-                                           PeriodProperties)
+from pandas.core.indexes.accessors import (
+    CombinedDatetimelikeProperties,
+    DatetimeProperties,
+    PeriodProperties,
+)
 from pandas.core.strings import StringMethods
 from pandas.util._decorators import doc
 
-_str_boolean_methods = set([
-    'contains',
-    'endswith',
-    'isalnum',
-    'isalpha',
-    'isdecimal',
-    'isdigit',
-    'islower',
-    'isnumeric',
-    'isspace',
-    'istitle',
-    'isupper',
-    'match',
-    'startswith',
-])
-
-_date_boolean_methods = set([
-    'is_leap_year',
-    'is_month_end',
-    'is_month_start',
-    'is_quarter_end',
-    'is_quarter_start',
-    'is_year_end',
-    'is_year_start',
-])
+_str_boolean_methods = set(
+    [
+        'contains',
+        'endswith',
+        'isalnum',
+        'isalpha',
+        'isdecimal',
+        'isdigit',
+        'islower',
+        'isnumeric',
+        'isspace',
+        'istitle',
+        'isupper',
+        'match',
+        'startswith',
+    ]
+)
+
+_date_boolean_methods = set(
+    [
+        'is_leap_year',
+        'is_month_end',
+        'is_month_start',
+        'is_quarter_end',
+        'is_quarter_start',
+        'is_year_end',
+        'is_year_start',
+    ]
+)
 
 
 class StringSelectMethods(StringMethods):
@@ -47,11 +53,14 @@ def __init__(self, *args, **kwargs):
         super().__init__(self._series, *args[1:], **kwargs)
 
     def __getattribute__(self, attr):
-        if (not attr.startswith("_") and  # noqa
-                inspect.isroutine(getattr(StringMethods, attr, None))
-            and attr not in _str_boolean_methods):  # noqa
-            raise NotImplementedError("Boolean selection with this method "
-                                      "does not make sense.")
+        if (
+            not attr.startswith("_")
+            and inspect.isroutine(getattr(StringMethods, attr, None))  # noqa
+            and attr not in _str_boolean_methods
+        ):  # noqa
+            raise NotImplementedError(
+                "Boolean selection with this method " "does not make sense."
+            )
         else:
             return super().__getattribute__(attr)
 
@@ -69,11 +78,16 @@ def __init__(self, parent, *args, **kwargs):
         super().__init__(*args, **kwargs)
 
     def __getattribute__(self, attr):
-        if (not attr.startswith("_") and  # noqa
-                inspect.isroutine(getattr(DatetimeProperties, attr, None))
-            and attr not in _date_boolean_methods): # noqa
-            raise NotImplementedError("Boolean selection with this method "
-                                      "does not make sense.")
+        if (
+            not attr.startswith("_")
+            and inspect.isroutine(  # noqa
+                getattr(DatetimeProperties, attr, None)
+            )
+            and attr not in _date_boolean_methods
+        ):  # noqa
+            raise NotImplementedError(
+                "Boolean selection with this method " "does not make sense."
+            )
         elif attr in _date_boolean_methods:
             idx = super().__getattribute__(attr)
             return self._parent_frame.loc[idx]
@@ -96,17 +110,16 @@ def __new__(cls, series):
         properties = super().__new__(cls, series._series)
         if isinstance(properties, DatetimeProperties):
             return SelectDatetimeProperties(
-                series._parent,
-                properties._parent,
-                properties.orig)
+                series._parent, properties._parent, properties.orig
+            )
         elif isinstance(properties, PeriodProperties):
             return SelectPeriodProperties(
-                series._frame,
-                properties._parent,
-                properties.orig
+                series._frame, properties._parent, properties.orig
             )
-        raise AttributeError("Can only use select.dt accessor on"
-                             "datetimelike and periodlike values.")
+        raise AttributeError(
+            "Can only use select.dt accessor on"
+            "datetimelike and periodlike values."
+        )
 
 
 def selector_wrapper(klass, method_name):
@@ -119,6 +132,7 @@ def selector(self, *args, **kwargs):
         series = self._series
         idx = getattr(klass, method_name)(series, *args, **kwargs)
         return self._parent.loc[idx]
+
     return selector
 
 

pandas_select/tests/conftest.py

@@ -5,9 +5,11 @@
 
 @pytest.fixture
 def dta():
-    return pd.DataFrame.from_dict({
-        'A': np.arange(1, 16),
-        'B': pd.date_range('2020-01-01', periods=15),
-        'C': ['A', 'B', 'C'] * 5,
-        'D': pd.Categorical(['A', 'B', 'C'] * 5)
-    })
+    return pd.DataFrame.from_dict(
+        {
+            'A': np.arange(1, 16),
+            'B': pd.date_range('2020-01-01', periods=15),
+            'C': ['A', 'B', 'C'] * 5,
+            'D': pd.Categorical(['A', 'B', 'C'] * 5),
+        }
+    )

pandas_select/tests/test_select.py

@@ -49,11 +49,11 @@ def test_str(dta):
 def test_dt(dta):
     assert dta.select.B.dt.is_leap_year.equals(dta)
     assert len(dta.select.B.dt.is_month_end) == 0
-    assert (dta.select.B.dt.is_month_start.equals(dta.loc[[0]]))
+    assert dta.select.B.dt.is_month_start.equals(dta.loc[[0]])
     assert len(dta.select.B.dt.is_quarter_end) == 0
-    assert (dta.select.B.dt.is_quarter_start.equals(dta.loc[[0]]))
+    assert dta.select.B.dt.is_quarter_start.equals(dta.loc[[0]])
     assert len(dta.select.B.dt.is_year_end) == 0
-    assert (dta.select.B.dt.is_year_start.equals(dta.loc[[0]]))
+    assert dta.select.B.dt.is_year_start.equals(dta.loc[[0]])
 
 
 def test_index(dta):

pyproject.toml

@@ -0,0 +1,18 @@
+[tool.black]
+line-length = 79
+skip-string-normalization = true
+exclude = '''(
+  /(
+      \.eggs         # exclude a few common directories in the
+    | \.git          # root of the project
+    | \.hg
+    | \.mypy_cache
+    | \.tox
+    | \.venv
+    | _build
+    | buck-out
+    | build
+    | dist
+  )/
+)
+'''

setup.py

@@ -4,5 +4,5 @@
     name='pandas_select',
     version='1.0',
     description='Add a select accessor to pandas',
-    packages=['pandas_select']
+    packages=['pandas_select'],
 )