File tree 1 file changed +7
-11
lines changed
1 file changed +7
-11
lines changed Original file line number Diff line number Diff line change @@ -15,17 +15,13 @@ class TestWB(tm.TestCase):
15
15
@network
16
16
def test_wdi_search (self ):
17
17
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
+ GDP_labels = [l for l in result .name if 'GDP' in l .upper ()]
24
+ self .assertTrue (len (GDP_labels ) > 1 )
29
25
30
26
@slow
31
27
@network
You can’t perform that action at this time.
0 commit comments