Merge pull request #181 from gliptak/chopdata

ENH: Set chopdata to private in Yahoo options

Author: femtotrader

.gitignore

@@ -4,4 +4,5 @@ __pycache__
 .coverage
 build
 dist
-docs/build
+docs/build
+.tox/

pandas_datareader/tests/test_yahoo_options.py

@@ -124,18 +124,18 @@ def test_sample_page_price_quote_time1(self):
 
     def test_chop(self):
         # regression test for #7625
-        self.aapl.chop_data(self.data1, above_below=2, underlying_price=np.nan)
-        chopped = self.aapl.chop_data(self.data1, above_below=2, underlying_price=100)
+        self.aapl._chop_data(self.data1, above_below=2, underlying_price=np.nan)
+        chopped = self.aapl._chop_data(self.data1, above_below=2, underlying_price=100)
         self.assertTrue(isinstance(chopped, DataFrame))
         self.assertTrue(len(chopped) > 1)
-        chopped2 = self.aapl.chop_data(self.data1, above_below=2, underlying_price=None)
+        chopped2 = self.aapl._chop_data(self.data1, above_below=2, underlying_price=None)
         self.assertTrue(isinstance(chopped2, DataFrame))
         self.assertTrue(len(chopped2) > 1)
 
     def test_chop_out_of_strike_range(self):
         # regression test for #7625
-        self.aapl.chop_data(self.data1, above_below=2, underlying_price=np.nan)
-        chopped = self.aapl.chop_data(self.data1, above_below=2, underlying_price=100000)
+        self.aapl._chop_data(self.data1, above_below=2, underlying_price=np.nan)
+        chopped = self.aapl._chop_data(self.data1, above_below=2, underlying_price=100000)
         self.assertTrue(isinstance(chopped, DataFrame))
         self.assertTrue(len(chopped) > 1)
 

pandas_datareader/yahoo/options.py

@@ -406,9 +406,9 @@ def get_near_stock_price(self, above_below=2, call=True, put=False,
         """
         expiry = self._try_parse_dates(year, month, expiry)
         data = self._get_data_in_date_range(expiry, call=call, put=put)
-        return self.chop_data(data, above_below, self.underlying_price)
+        return self._chop_data(data, above_below, self.underlying_price)
 
-    def chop_data(self, df, above_below=2, underlying_price=None):
+    def _chop_data(self, df, above_below=2, underlying_price=None):
         """Returns a data frame only options that are near the current stock price."""
 
         if not underlying_price:
@@ -552,7 +552,7 @@ def get_forward_data(self, months, call=True, put=False, near=False,
         dates = (date for date in self.expiry_dates if date <= end_date.date())
         data = self._get_data_in_date_range(dates, call=call, put=put)
         if near:
-            data = self.chop_data(data, above_below=above_below)
+            data = self._chop_data(data, above_below=above_below)
         return data
 
     def get_all_data(self, call=True, put=True):