File tree Expand file tree Collapse file tree 1 file changed +6
-11
lines changed
Expand file tree Collapse file tree 1 file changed +6
-11
lines changed Original file line number Diff line number Diff line change @@ -15,17 +15,12 @@ class TestWB(tm.TestCase):
1515 @network
1616 def test_wdi_search (self ):
1717
18- expected = {u ('id' ): {6716 : u ('NY.GDP.PCAP.KD' ),
19- 6718 : u ('NY.GDP.PCAP.KN' ),
20- 6720 : u ('NY.GDP.PCAP.PP.KD' )},
21- u ('name' ): {6716 : u ('GDP per capita (constant 2005 US$)' ),
22- 6718 : u ('GDP per capita (constant LCU)' ),
23- 6720 : u ('GDP per capita, PPP (constant 2011 '
24- 'international $)' )}}
25- result = search ('gdp.*capita.*constant' ).loc [6716 :,['id' ,'name' ]]
26- expected = pandas .DataFrame (expected )
27- expected .index = result .index
28- assert_frame_equal (result , expected )
18+ # Test that a name column exists, and that some results were returned
19+ # ...without being too strict about what the actual contents of the
20+ # results actually are. The fact that there are some, is good enough.
21+
22+ result = search ('gdp.*capita.*constant' )
23+ self .assertTrue (result .name .str .contains ('GDP' ).any ())
2924
3025 @slow
3126 @network
You can’t perform that action at this time.
0 commit comments