Merge pull request #456 from bashtage/update-world-bank

CLN: Update WorldBank endpoint

Author: bashtage

pandas_datareader/tests/test_wb.py

@@ -210,7 +210,7 @@ def test_wdi_get_indicators(self):
 
         for result in [result1, result2, result3, result4]:
             exp_col = pd.Index(['id', 'name', 'source', 'sourceNote',
-                                'sourceOrganization', 'topics'])
+                                'sourceOrganization', 'topics', 'unit'])
             # assert_index_equal doesn't exists
             assert result.columns.equals(exp_col)
             assert len(result) > 10000

pandas_datareader/wb.py

@@ -14,6 +14,8 @@
 # 3-digit ISO 3166-1 alpha-3, codes, with 'all', 'ALL', and 'All' appended ot
 # the end.
 
+WB_API_URL = 'https://api.worldbank.org/v2'
+
 country_codes = ['AD', 'AE', 'AF', 'AG', 'AI', 'AL', 'AM', 'AO', 'AQ', 'AR',
                  'AS', 'AT', 'AU', 'AW', 'AX', 'AZ', 'BA', 'BB', 'BD', 'BE',
                  'BF', 'BG', 'BH', 'BI', 'BJ', 'BL', 'BM', 'BN', 'BO', 'BQ',
@@ -158,8 +160,7 @@ def __init__(self, symbols=None, countries=None,
     @property
     def url(self):
         countries = ';'.join(self.countries)
-        return ("http://api.worldbank.org/countries/" +
-                countries + "/indicators/")
+        return WB_API_URL + '/countries/' + countries + '/indicators/'
 
     @property
     def params(self):
@@ -251,7 +252,7 @@ def get_countries(self):
             country code, region, income level,
             capital city, latitude, and longitude
         """
-        url = 'http://api.worldbank.org/countries/?per_page=1000&format=json'
+        url = WB_API_URL + '/countries/?per_page=1000&format=json'
 
         resp = self._get_response(url)
         data = resp.json()[1]
@@ -274,7 +275,7 @@ def get_indicators(self):
         if isinstance(_cached_series, pd.DataFrame):
             return _cached_series.copy()
 
-        url = 'http://api.worldbank.org/indicators?per_page=50000&format=json'
+        url = WB_API_URL + '/indicators?per_page=50000&format=json'
 
         resp = self._get_response(url)
         data = resp.json()[1]