1
- import os
2
-
3
1
import pytest
4
2
from requests .exceptions import HTTPError
5
3
6
4
import pandas_datareader as pdr
7
5
import pandas_datareader .data as web
6
+ from pandas_datareader .exceptions import ImmediateDeprecationError
8
7
9
8
pytestmark = pytest .mark .requires_api_key
10
9
11
- TEST_API_KEY = os . getenv ( "ENIGMA_API_KEY" )
10
+ TEST_API_KEY = "DEPRECATED"
12
11
13
12
14
- @pytest .mark .skipif (TEST_API_KEY is None , reason = "no enigma_api_key" )
15
13
class TestEnigma (object ):
16
14
@property
17
15
def dataset_id (self ):
@@ -28,24 +26,24 @@ def setup_class(cls):
28
26
29
27
def test_enigma_datareader (self ):
30
28
try :
31
- df = web . DataReader ( self . dataset_id , "enigma" , api_key = TEST_API_KEY )
32
- assert "case_number" in df . columns
29
+ with pytest . raises ( ImmediateDeprecationError ):
30
+ web . DataReader ( self . dataset_id , "enigma" , api_key = TEST_API_KEY )
33
31
except HTTPError as e :
34
32
pytest .skip (e )
35
33
36
34
def test_enigma_get_data_enigma (self ):
37
35
try :
38
- df = pdr . get_data_enigma ( self . dataset_id , TEST_API_KEY )
39
- assert "case_number" in df . columns
36
+ with pytest . raises ( ImmediateDeprecationError ):
37
+ pdr . get_data_enigma ( self . dataset_id , TEST_API_KEY )
40
38
except HTTPError as e :
41
39
pytest .skip (e )
42
40
43
41
def test_bad_key (self ):
44
- with pytest .raises (HTTPError ):
42
+ with pytest .raises (ImmediateDeprecationError ):
45
43
web .DataReader (self .dataset_id , "enigma" , api_key = TEST_API_KEY + "xxx" )
46
44
47
45
def test_bad_dataset_id (self ):
48
- with pytest .raises (HTTPError ):
46
+ with pytest .raises (ImmediateDeprecationError ):
49
47
web .DataReader (
50
48
"zzzzzzzz-zzzz-zzzz-zzzz-zzzzzzzzzzz" , "enigma" , api_key = TEST_API_KEY
51
49
)
0 commit comments