Skip to content

Commit bd052e4

Browse files
Merge pull request #38 from dstephens99/TestGoogle
TST: Fix tests in TestGoogle.
2 parents 7adaca5 + 3b4c3a7 commit bd052e4

File tree

1 file changed

+4
-4
lines changed

1 file changed

+4
-4
lines changed

pandas_datareader/tests/test_data.py

+4-4
Original file line numberDiff line numberDiff line change
@@ -38,7 +38,7 @@ def assert_n_failed_equals_n_null_columns(wngs, obj, cls=SymbolWarning):
3838
all_nan_cols = pd.Series(dict((k, pd.isnull(v).all()) for k, v in
3939
compat.iteritems(obj)))
4040
n_all_nan_cols = all_nan_cols.sum()
41-
valid_warnings = pd.Series([wng for wng in wngs if isinstance(wng, cls)])
41+
valid_warnings = pd.Series([wng for wng in wngs if wng.category == cls])
4242
assert_equal(len(valid_warnings), n_all_nan_cols)
4343
failed_symbols = all_nan_cols[all_nan_cols].index
4444
msgs = valid_warnings.map(lambda x: x.message)
@@ -81,17 +81,17 @@ def test_get_goog_volume(self):
8181
for locale in self.locales:
8282
with tm.set_locale(locale):
8383
df = web.get_data_google('GOOG').sort_index()
84-
self.assertEqual(df.Volume.ix['OCT-08-2010'], 2863473)
84+
self.assertEqual(df.Volume.ix['JAN-02-2015'], 1446662)
8585

8686
def test_get_multi1(self):
8787
for locale in self.locales:
8888
sl = ['AAPL', 'AMZN', 'GOOG']
8989
with tm.set_locale(locale):
9090
pan = web.get_data_google(sl, '2012')
91-
ts = pan.Close.GOOG.index[pan.Close.AAPL > pan.Close.GOOG]
91+
ts = pan.Close.GOOG.index[pan.Close.AAPL < pan.Close.GOOG]
9292
if (hasattr(pan, 'Close') and hasattr(pan.Close, 'GOOG') and
9393
hasattr(pan.Close, 'AAPL')):
94-
self.assertEqual(ts[0].dayofyear, 96)
94+
self.assertEqual(ts[0].dayofyear, 3)
9595
else:
9696
self.assertRaises(AttributeError, lambda: pan.Close)
9797

0 commit comments

Comments
 (0)