Added test for wb.get_countries

Author: clarkfitzg

doc/source/release.rst

@@ -102,6 +102,7 @@ Improvements to existing features
 
 Bug Fixes
 ~~~~~~~~~
+  - Bug in ``io.wb.get_countries`` not including all countries (:issue:`6008`)
   - Bug in Series replace with timestamp dict (:issue:`5797`)
   - read_csv/read_table now respects the `prefix` kwarg (:issue:`5732`).
   - Bug in selection with missing values via ``.ix`` from a duplicate indexed DataFrame failing (:issue:`5835`)

pandas/io/tests/test_wb.py

@@ -5,9 +5,10 @@
 from pandas.util.testing import network
 from pandas.util.testing import assert_frame_equal
 from numpy.testing.decorators import slow
-from pandas.io.wb import search, download
+from pandas.io.wb import search, download, get_countries
 import pandas.util.testing as tm
 
+
 class TestWB(tm.TestCase):
 
     @slow
@@ -30,7 +31,6 @@ def test_wdi_search(self):
         expected.index = result.index
         assert_frame_equal(result, expected)
 
-
     @slow
     @network
     def test_wdi_download(self):
@@ -43,6 +43,12 @@ def test_wdi_download(self):
         expected.index = result.index
         assert_frame_equal(result, pandas.DataFrame(expected))
 
+    @slow
+    @network
+    def test_wdi_get_countries(self):
+        result = get_countries()
+        self.assertTrue('Zimbabwe' in list(result['name']))
+
 
 if __name__ == '__main__':
     nose.runmodule(argv=[__file__, '-vvs', '-x', '--pdb', '--pdb-failure'],