@@ -26,13 +26,17 @@ def test_infer_start_end_from_symbols(self):
26
26
assert df .index [0 ].year == 2010
27
27
assert df .index [- 1 ].year == 2018
28
28
29
- tickers = [f'{ sec } { geo } ' for sec in ['RGDP' , 'CPI' , 'URATE' ]
30
- for geo in ['US' , 'UK' , 'ES' , 'AR' ]]
29
+ tickers = [
30
+ f"{ sec } { geo } "
31
+ for sec in ["RGDP" , "CPI" , "URATE" ]
32
+ for geo in ["US" , "UK" , "ES" , "AR" ]
33
+ ]
31
34
32
35
@pytest .mark .parametrize ("ticker" , tickers )
33
36
def test_fetch_single_ticker_series (self , ticker ):
34
37
df = web .DataReader (
35
- f"ticker={ ticker } " , "econdb" ,
38
+ f"ticker={ ticker } " ,
39
+ "econdb" ,
36
40
start = pd .Timestamp ("2010-01-01" ),
37
41
end = pd .Timestamp ("2013-01-27" ),
38
42
)
@@ -52,35 +56,39 @@ def test_single_nonticker_series(self):
52
56
53
57
def test_filtered_dataset (self ):
54
58
df = web .DataReader (
55
- '&' .join ([
56
- 'dataset=PRC_HICP_MIDX' ,
57
- 'v=Geopolitical entity (reporting)' ,
58
- 'h=TIME' ,
59
- 'from=2022-03-01' ,
60
- 'to=2022-09-01' ,
61
- 'COICOP=[CP00]' ,
62
- 'FREQ=[M]' ,
63
- 'GEO=[ES,AT,CZ,IT,CH]' ,
64
- 'UNIT=[I15]'
65
- ]),
66
- 'econdb'
59
+ "&" .join (
60
+ [
61
+ "dataset=PRC_HICP_MIDX" ,
62
+ "v=Geopolitical entity (reporting)" ,
63
+ "h=TIME" ,
64
+ "from=2022-03-01" ,
65
+ "to=2022-09-01" ,
66
+ "COICOP=[CP00]" ,
67
+ "FREQ=[M]" ,
68
+ "GEO=[ES,AT,CZ,IT,CH]" ,
69
+ "UNIT=[I15]" ,
70
+ ]
71
+ ),
72
+ "econdb" ,
67
73
)
68
74
assert df .shape [1 ] == 5
69
75
assert isinstance (df .index , pd .DatetimeIndex )
70
76
71
77
def test_australia_gdp (self ):
72
78
df = web .DataReader (
73
- '&' .join ([
74
- 'dataset=ABS_GDP' ,
75
- '4=[7]' ,
76
- '6=[11]' ,
77
- '16=[1267]' ,
78
- 'v=TIME' ,
79
- 'h=Indicator' ,
80
- 'from=2019-10-01' ,
81
- 'to=2022-06-01' ,
82
- 'GEO=[13]'
83
- ]),
84
- 'econdb'
79
+ "&" .join (
80
+ [
81
+ "dataset=ABS_GDP" ,
82
+ "4=[7]" ,
83
+ "6=[11]" ,
84
+ "16=[1267]" ,
85
+ "v=TIME" ,
86
+ "h=Indicator" ,
87
+ "from=2019-10-01" ,
88
+ "to=2022-06-01" ,
89
+ "GEO=[13]" ,
90
+ ]
91
+ ),
92
+ "econdb" ,
85
93
)
86
94
assert_equal (df .squeeze ().loc ["2020-10-01" ], 508603 )
0 commit comments