1- import os
2-
31import pytest
42from requests .exceptions import HTTPError
53
64import pandas_datareader as pdr
75import pandas_datareader .data as web
6+ from pandas_datareader .exceptions import ImmediateDeprecationError
87
98pytestmark = pytest .mark .requires_api_key
109
11- TEST_API_KEY = os . getenv ( "ENIGMA_API_KEY" )
10+ TEST_API_KEY = "DEPRECATED"
1211
1312
14- @pytest .mark .skipif (TEST_API_KEY is None , reason = "no enigma_api_key" )
1513class TestEnigma (object ):
1614 @property
1715 def dataset_id (self ):
@@ -28,24 +26,24 @@ def setup_class(cls):
2826
2927 def test_enigma_datareader (self ):
3028 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 )
3331 except HTTPError as e :
3432 pytest .skip (e )
3533
3634 def test_enigma_get_data_enigma (self ):
3735 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 )
4038 except HTTPError as e :
4139 pytest .skip (e )
4240
4341 def test_bad_key (self ):
44- with pytest .raises (HTTPError ):
42+ with pytest .raises (ImmediateDeprecationError ):
4543 web .DataReader (self .dataset_id , "enigma" , api_key = TEST_API_KEY + "xxx" )
4644
4745 def test_bad_dataset_id (self ):
48- with pytest .raises (HTTPError ):
46+ with pytest .raises (ImmediateDeprecationError ):
4947 web .DataReader (
5048 "zzzzzzzz-zzzz-zzzz-zzzz-zzzzzzzzzzz" , "enigma" , api_key = TEST_API_KEY
5149 )
0 commit comments