@@ -19,36 +19,33 @@ def setUpClass(cls):
1919 super (TestEnigma , cls ).setUpClass ()
2020 _skip_if_no_lxml ()
2121
22- def setUp (self ):
23- raise nose .SkipTest ()
22+ def test_enigma_datareader (self ):
23+ try :
24+ df = web .DataReader ('enigma.inspections.restaurants.fl' ,
25+ 'enigma' , access_key = TEST_API_KEY )
26+ self .assertTrue ('serialid' in df .columns )
27+ except HTTPError as e : # pragma: no cover
28+ raise nose .SkipTest (e )
2429
25- def test_enigma (self ):
26- self . assertTrue ( 'serialid' in list (
27- web . DataReader ( 'enigma.inspections.restaurants.fl' ,
28- 'enigma' ,
29- access_key = TEST_API_KEY ) .columns ) )
30- self . assertTrue ( 'serialid' in list ( pdr . get_data_enigma (
31- 'enigma.inspections.restaurants.fl' , TEST_API_KEY )) )
30+ def test_enigma_get_data_enigma (self ):
31+ try :
32+ df = pdr . get_data_enigma (
33+ 'enigma.inspections.restaurants.fl' , TEST_API_KEY )
34+ self . assertTrue ( 'serialid' in df .columns )
35+ except HTTPError as e : # pragma: no cover
36+ raise nose . SkipTest ( e )
3237
3338 def test_bad_key (self ):
34- _exception = None
35- try :
39+ with tm .assertRaises (HTTPError ):
3640 web .DataReader ('enigma.inspections.restaurants.fl' ,
3741 'enigma' ,
3842 access_key = TEST_API_KEY + 'xxx' )
39- except HTTPError as e :
40- _exception = e
41- assert isinstance (_exception , HTTPError )
4243
4344 def test_bad_url (self ):
44- _exception = None
45- try :
45+ with tm .assertRaises (HTTPError ):
4646 web .DataReader ('enigma.inspections.restaurants.fllzzy' ,
4747 'enigma' ,
4848 access_key = TEST_API_KEY )
49- except Exception as e :
50- _exception = e
51- assert isinstance (_exception , HTTPError )
5249
5350
5451if __name__ == '__main__' :
0 commit comments