def test_datareader_world_bank(): #print("="*5 + "Pandas original DataReader" + "="*5) #dat = wb.search('gdp.*capita.*const') #print(dat.iloc[:,:2]) #dat = wb.download(indicator=u'NY.GDP.PCAP.KD', country=['US', 'CA', 'MX'], start=2005, end=2008) #print(dat) #assert(isinstance(wb.country_codes, list)) print("=" * 5 + "New DataReader" + "=" * 5) world_bank = DataReader("WorldBank", expire_after=expire_after) assert (isinstance(world_bank.country_codes, list)) data = world_bank.search('gdp.*capita.*const') print(data.iloc[:, :2]) assert (isinstance(data, pd.DataFrame)) data = world_bank.download(indicator='NY.GDP.PCAP.KD', country=['US', 'CA', 'MX'], start=2005, end=2008) print(data) assert (isinstance(data, pd.DataFrame))