@@ -19,36 +19,33 @@ def setUpClass(cls):
19
19
super (TestEnigma , cls ).setUpClass ()
20
20
_skip_if_no_lxml ()
21
21
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 )
24
29
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 )
32
37
33
38
def test_bad_key (self ):
34
- _exception = None
35
- try :
39
+ with tm .assertRaises (HTTPError ):
36
40
web .DataReader ('enigma.inspections.restaurants.fl' ,
37
41
'enigma' ,
38
42
access_key = TEST_API_KEY + 'xxx' )
39
- except HTTPError as e :
40
- _exception = e
41
- assert isinstance (_exception , HTTPError )
42
43
43
44
def test_bad_url (self ):
44
- _exception = None
45
- try :
45
+ with tm .assertRaises (HTTPError ):
46
46
web .DataReader ('enigma.inspections.restaurants.fllzzy' ,
47
47
'enigma' ,
48
48
access_key = TEST_API_KEY )
49
- except Exception as e :
50
- _exception = e
51
- assert isinstance (_exception , HTTPError )
52
49
53
50
54
51
if __name__ == '__main__' :
0 commit comments