4
4
import pandas as pd
5
5
6
6
from pandas_datareader import data as web
7
+ from pandas_datareader ._utils import RemoteDataError
7
8
8
9
from datetime import datetime
9
10
@@ -32,7 +33,7 @@ def end(self):
32
33
@pytest .mark .skipif (TEST_API_KEY is None ,
33
34
reason = "ALPHAVANTAGE_API_KEY not set" )
34
35
def test_av_bad_symbol (self ):
35
- with pytest .raises (ValueError ):
36
+ with pytest .raises (( ValueError , RemoteDataError ) ):
36
37
web .DataReader ("BADTICKER" , "av-daily" , start = self .start ,
37
38
end = self .end )
38
39
@@ -42,7 +43,7 @@ def test_av_daily(self):
42
43
df = web .DataReader ("AAPL" , "av-daily" , start = self .start , end = self .end )
43
44
assert df .columns .equals (self .col_index )
44
45
assert len (df ) == 578
45
- assert df ["volume" ][- 1 ] == 19118319
46
+ assert df ["volume" ][- 1 ] == 19178000
46
47
47
48
expected1 = df .loc ["2017-02-09" ]
48
49
assert expected1 ["close" ] == 132.42
@@ -62,19 +63,17 @@ def test_av_daily_adjusted(self):
62
63
"dividend amount" ,
63
64
"split coefficient" ]))
64
65
assert len (df ) == 578
65
- assert df ["volume" ][- 1 ] == 19118319
66
+ assert df ["volume" ][- 1 ] == 19178000
66
67
67
68
expected1 = df .loc ["2017-02-09" ]
68
69
assert expected1 ["close" ] == 132.42
69
70
assert expected1 ["high" ] == 132.445
70
- assert expected1 ["adjusted close" ] == 130.3505
71
71
assert expected1 ["dividend amount" ] == 0.57
72
72
assert expected1 ["split coefficient" ] == 1.0
73
73
74
74
expected2 = df .loc ["2017-05-24" ]
75
75
assert expected2 ["close" ] == 153.34
76
76
assert expected2 ["high" ] == 154.17
77
- assert expected2 ["adjusted close" ] == 151.5612
78
77
assert expected2 ["dividend amount" ] == 0.00
79
78
assert expected2 ["split coefficient" ] == 1.0
80
79
@@ -84,15 +83,10 @@ def _helper_df_weekly_monthly(df, adj=False):
84
83
expected1 = df .loc ["2015-02-27" ]
85
84
assert expected1 ["close" ] == 128.46
86
85
assert expected1 ["high" ] == 133.60
87
- if adj :
88
- assert expected1 ["adjusted close" ] == 121.5859
89
86
90
87
expected2 = df .loc ["2017-03-31" ]
91
88
assert expected2 ["close" ] == 143.66
92
89
assert expected2 ["high" ] == 144.5
93
- if adj :
94
- assert expected2 ["adjusted close" ] == 141.4148
95
- assert expected2 ["dividend amount" ] == 0.00
96
90
97
91
@pytest .mark .skipif (TEST_API_KEY is None ,
98
92
reason = "ALPHAVANTAGE_API_KEY not set" )
0 commit comments