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