Skip to content

Commit ada53df

Browse files
authored
FIX FRED test
Closes #218
1 parent 6075667 commit ada53df

File tree

1 file changed

+13
-5
lines changed

1 file changed

+13
-5
lines changed

pandas_datareader/tests/test_fred.py

Lines changed: 13 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -19,15 +19,23 @@ def test_fred(self):
1919
# FRED.
2020

2121
start = datetime(2010, 1, 1)
22-
end = datetime(2013, 1, 27)
22+
end = datetime(2013, 1, 1)
2323

24-
received = web.DataReader("GDP", "fred", start, end)['GDP'].tail(1)[0]
24+
df = web.DataReader("GDP", "fred", start, end)
25+
ts = df['GDP']
26+
self.assertEqual(ts.index[0], pd.to_datetime("2010-01-01"))
27+
self.assertEqual(ts.index[-1], pd.to_datetime("2013-01-01"))
28+
self.assertEqual(ts.index.name, "DATE")
29+
self.assertEqual(ts.name, "GDP")
30+
received = ts.tail(1)[0]
2531

26-
# < 7/30/14 16535 was returned
32+
# < 2014-07-30 16535 was returned
2733
# self.assertEqual(int(received), 16535)
28-
# < 8/20/15 16502 was returned
34+
# < 2015-08-20 16502 was returned
2935
# self.assertEqual(int(received), 16502)
30-
self.assertEqual(int(received), 16440)
36+
# < 2016-08-01 16440 was returned
37+
# self.assertEqual(int(received), 16440)
38+
self.assertEqual(int(received), 16475)
3139

3240
with tm.assertRaises(RemoteDataError):
3341
web.DataReader("NON EXISTENT SERIES", 'fred', start, end)

0 commit comments

Comments
 (0)