Merge pull request #96 from davidastephens/improve_cover

TST: Improve coverage

Author: davidastephens

pandas_datareader/fred.py

@@ -36,7 +36,7 @@ def fetch_data(url, name):
                             na_values='.')
         try:
             return data.truncate(start, end)
-        except KeyError:
+        except KeyError: # pragma: no cover
             if data.ix[3].name[7:12] == 'Error':
                 raise IOError("Failed to get the data. Check that {0!r} is "
                               "a valid FRED series.".format(name))

pandas_datareader/tests/test_data.py

@@ -216,6 +216,12 @@ def test_get_data_single_symbol(self):
         # just test that we succeed
         web.get_data_yahoo('GOOG')
 
+    def test_get_data_adjust_price(self):
+        goog = web.get_data_yahoo('GOOG')
+        goog_adj = web.get_data_yahoo('GOOG', adjust_price=True)
+        self.assertTrue('Adj Close' not in goog_adj.columns)
+        self.assertTrue((goog['Open']*goog_adj['Adj_Ratio']).equals(goog_adj['Open']))
+
     def test_get_data_interval(self):
         # daily interval data
         pan = web.get_data_yahoo('XOM', '2013-01-01', '2013-12-31', interval='d')
@@ -404,6 +410,9 @@ def test_chop(self):
         chopped = self.aapl.chop_data(self.data1, above_below=2, underlying_price=100)
         self.assert_(isinstance(chopped, DataFrame))
         self.assertTrue(len(chopped) > 1)
+        chopped2 = self.aapl.chop_data(self.data1, above_below=2, underlying_price=None)
+        self.assert_(isinstance(chopped2, DataFrame))
+        self.assertTrue(len(chopped2) > 1)
 
     def test_chop_out_of_strike_range(self):
         #regression test for #7625

pandas_datareader/yahoo/components.py

@@ -4,7 +4,7 @@
 _URL = 'http://download.finance.yahoo.com/d/quotes.csv?'
 
 
-def _get_data(idx_sym):
+def _get_data(idx_sym): # pragma: no cover
     """
     Returns DataFrame containing list of component information for
     index represented in idx_sym from yahoo. Includes component symbol

pandas_datareader/yahoo/options.py

@@ -150,7 +150,7 @@ def _yahoo_url_from_expiry(self, expiry):
         try:
             expiry_links = self._expiry_links
 
-        except AttributeError:
+        except AttributeError: # pragma: no cover
             _, expiry_links = self._get_expiry_dates_and_links()
 
         return self._FINANCE_BASE_URL + expiry_links[expiry]
@@ -478,7 +478,7 @@ def _validate_expiry(self, expiry):
             return index[index.date >= expiry][0].date()
 
     def get_forward_data(self, months, call=True, put=False, near=False,
-                         above_below=2):
+            above_below=2): # pragma: no cover
         """
         ***Experimental***
         Gets either call, put, or both data for months starting in the current
@@ -624,14 +624,14 @@ def _get_expiry_dates_and_links(self):
 
         try:
             links = root.xpath('//*[@id="options_menu"]/form/select/option')
-        except IndexError:
+        except IndexError: # pragma: no cover
             raise RemoteDataError('Expiry dates not available')
 
         expiry_dates = [dt.datetime.strptime(element.text, "%B %d, %Y").date() for element in links]
         links = [element.attrib['data-selectbox-link'] for element in links]
 
         if len(expiry_dates) == 0:
-            raise RemoteDataError('Data not available')
+            raise RemoteDataError('Data not available') # pragma: no cover
 
         expiry_links = dict(zip(expiry_dates, links))
         self._expiry_links = expiry_links
@@ -645,17 +645,17 @@ def _parse_url(self, url):
         """
         try:
             from lxml.html import parse
-        except ImportError:
+        except ImportError: # pragma: no cover
             raise ImportError("Please install lxml if you want to use the "
                               "{0!r} class".format(self.__class__.__name__))
         try:
             doc = parse(url)
-        except _network_error_classes:
+        except _network_error_classes: # pragma: no cover
             raise RemoteDataError("Unable to parse URL "
                                   "{0!r}".format(url))
         else:
             root = doc.getroot()
-            if root is None:
+            if root is None: # pragma: no cover
                 raise RemoteDataError("Parsed URL {0!r} has no root"
                                       "element".format(url))
         return root
@@ -678,7 +678,7 @@ def _process_data(self, frame, type):
         try:
             frame['Underlying_Price'] = self.underlying_price
             frame["Quote_Time"] = self.quote_time
-        except AttributeError:
+        except AttributeError: # pragma: no cover
             frame['Underlying_Price'] = np.nan
             frame["Quote_Time"] = np.nan
         frame.rename(columns={'Open Int': 'Open_Int'}, inplace=True)