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,15 +33,15 @@ def end(self):
32
33
return datetime (2017 , 5 , 24 )
33
34
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
39
40
def test_av_daily (self ):
40
41
df = web .DataReader ("AAPL" , "av-daily" , start = self .start , end = self .end )
41
42
assert df .columns .equals (self .col_index )
42
43
assert len (df ) == 578
43
- assert df ["volume" ][- 1 ] == 19118319
44
+ assert df ["volume" ][- 1 ] == 19178000
44
45
45
46
expected1 = df .loc ["2017-02-09" ]
46
47
assert expected1 ["close" ] == 132.42
@@ -58,19 +59,17 @@ def test_av_daily_adjusted(self):
58
59
"dividend amount" ,
59
60
"split coefficient" ]))
60
61
assert len (df ) == 578
61
- assert df ["volume" ][- 1 ] == 19118319
62
+ assert df ["volume" ][- 1 ] == 19178000
62
63
63
64
expected1 = df .loc ["2017-02-09" ]
64
65
assert expected1 ["close" ] == 132.42
65
66
assert expected1 ["high" ] == 132.445
66
- assert expected1 ["adjusted close" ] == 130.3505
67
67
assert expected1 ["dividend amount" ] == 0.57
68
68
assert expected1 ["split coefficient" ] == 1.0
69
69
70
70
expected2 = df .loc ["2017-05-24" ]
71
71
assert expected2 ["close" ] == 153.34
72
72
assert expected2 ["high" ] == 154.17
73
- assert expected2 ["adjusted close" ] == 151.5612
74
73
assert expected2 ["dividend amount" ] == 0.00
75
74
assert expected2 ["split coefficient" ] == 1.0
76
75
@@ -80,15 +79,10 @@ def _helper_df_weekly_monthly(df, adj=False):
80
79
expected1 = df .loc ["2015-02-27" ]
81
80
assert expected1 ["close" ] == 128.46
82
81
assert expected1 ["high" ] == 133.60
83
- if adj :
84
- assert expected1 ["adjusted close" ] == 121.5859
85
82
86
83
expected2 = df .loc ["2017-03-31" ]
87
84
assert expected2 ["close" ] == 143.66
88
85
assert expected2 ["high" ] == 144.5
89
- if adj :
90
- assert expected2 ["adjusted close" ] == 141.4148
91
- assert expected2 ["dividend amount" ] == 0.00
92
86
93
87
def test_av_weekly (self ):
94
88
df = web .DataReader ("AAPL" , "av-weekly" , start = self .start ,
0 commit comments