Satisfied linters.

Author: ericmjl

environment.yml

@@ -8,5 +8,6 @@ dependencies:
   - pip
   - pre-commit
   - pytest
+  - black
   - pip:
       - lazy-loader

tests/test_pandas_register.py

@@ -0,0 +1,41 @@
+"""Tests for pandas series and dataframe method registration."""
+import pandas_flavor as pf
+import pandas as pd
+
+
+def test_register_dataframe_method():
+    """Test register_dataframe_method."""
+
+    @pf.register_dataframe_method
+    def dummy_func(df: pd.DataFrame) -> pd.DataFrame:
+        """Dummy function.
+
+        Args:
+            df: a pandas DataFrame
+
+        Returns:
+            df: A pandas DataFrame.
+        """
+        return df
+
+    df = pd.DataFrame()
+    df.dummy_func()
+
+
+def test_register_series_method():
+    """Test register_series_method."""
+
+    @pf.register_series_method
+    def dummy_func(s: pd.Series) -> pd.Series:
+        """Dummy func.
+
+        Args:
+            s: A pandas Series.
+
+        Returns:
+            s: A pandas Series.
+        """
+        return s
+
+    ser = pd.Series()
+    ser.dummy_func()