@@ -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