1
1
from datetime import date , timedelta
2
+
2
3
import pytest
3
4
4
- from pandas_datareader ._utils import RemoteDataError
5
5
import pandas_datareader .data as web
6
+ from pandas_datareader ._utils import RemoteDataError
6
7
7
8
8
9
class TestBankOfCanada (object ):
9
-
10
+
10
11
@staticmethod
11
12
def get_symbol (currency_code , inverted = False ):
12
13
if inverted :
@@ -15,23 +16,27 @@ def get_symbol(currency_code, inverted=False):
15
16
return 'FX{}CAD' .format (currency_code )
16
17
17
18
def check_bankofcanada_count (self , code ):
18
- df = web .DataReader (self .get_symbol (code ), 'bankofcanada' , date .today () - timedelta (days = 30 ), date .today ())
19
+ df = web .DataReader (self .get_symbol (code ), 'bankofcanada' ,
20
+ date .today () - timedelta (days = 30 ), date .today ())
19
21
assert df .size > 15
20
22
21
23
def check_bankofcanada_valid (self , code ):
22
24
symbol = self .get_symbol (code )
23
- df = web .DataReader (symbol , 'bankofcanada' , date .today () - timedelta (days = 30 ), date .today ())
25
+ df = web .DataReader (symbol , 'bankofcanada' ,
26
+ date .today () - timedelta (days = 30 ), date .today ())
24
27
assert symbol in df .columns
25
28
26
29
def check_bankofcanada_inverted (self , code ):
27
30
symbol = self .get_symbol (code )
28
31
symbol_inverted = self .get_symbol (code , inverted = True )
29
32
30
- df = web .DataReader (symbol , 'bankofcanada' , date .today () - timedelta (days = 30 ), date .today ())
31
- df_inv = web .DataReader (symbol_inverted , 'bankofcanada' , date .today () - timedelta (days = 30 ), date .today ())
33
+ df = web .DataReader (symbol , 'bankofcanada' ,
34
+ date .today () - timedelta (days = 30 ), date .today ())
35
+ df_i = web .DataReader (symbol_inverted , 'bankofcanada' ,
36
+ date .today () - timedelta (days = 30 ), date .today ())
32
37
33
- pairs = zip ((1 / df )[symbol ].tolist (), df_inv [symbol_inverted ].tolist ())
34
- assert all (a - b < 0.01 for a , b in pairs )
38
+ pairs = zip ((1 / df )[symbol ].tolist (), df_i [symbol_inverted ].tolist ())
39
+ assert all (a - b < 0.01 for a , b in pairs )
35
40
36
41
def test_bankofcanada_usd_count (self ):
37
42
self .check_bankofcanada_count ('USD' )
@@ -53,8 +58,10 @@ def test_bankofcanada_eur_inverted(self):
53
58
54
59
def test_bankofcanada_bad_range (self ):
55
60
with pytest .raises (RemoteDataError ):
56
- web .DataReader ('FXCADUSD' , 'bankofcanada' , date .today (), date .today () - timedelta (days = 30 ))
61
+ web .DataReader ('FXCADUSD' , 'bankofcanada' ,
62
+ date .today (), date .today () - timedelta (days = 30 ))
57
63
58
64
def test_bankofcanada_bad_url (self ):
59
65
with pytest .raises (RemoteDataError ):
60
- web .DataReader ('abcdefgh' , 'bankofcanada' , date .today (), date .today () - timedelta (days = 30 ))
66
+ web .DataReader ('abcdefgh' , 'bankofcanada' ,
67
+ date .today () - timedelta (days = 30 ), date .today ())
0 commit comments