-
Notifications
You must be signed in to change notification settings - Fork 136
/
test_refdata.py
59 lines (47 loc) · 1.61 KB
/
test_refdata.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
from datetime import datetime
import pandas as pd
from iexfinance.refdata import (
get_symbols,
get_iex_symbols,
get_us_trading_dates_holidays,
get_region_symbols,
get_exchange_symbols,
)
IEX_MSG = (
"These functions return data for IEX listed symbols only. There is "
"only 1 listed IEX symbol."
)
class TestRef(object):
def setup_class(self):
self.keys = {"RecordID", "DailyListTimestamp", "CompanyName"}
self.start = datetime(2017, 5, 4)
def test_get_symbols(self):
d = get_symbols()
assert isinstance(d, pd.DataFrame)
def test_get_iex_symbols(self):
d = get_iex_symbols()
assert isinstance(d, pd.DataFrame)
def test_get_region_symbols(self):
d = get_region_symbols("ca")
assert isinstance(d, pd.DataFrame)
def test_get_exchange_symbols(self):
d = get_exchange_symbols("tse")
assert isinstance(d, pd.DataFrame)
def test_get_us_trading_dates_holidays(self):
assert isinstance(get_us_trading_dates_holidays("trade", "last"), pd.DataFrame)
assert isinstance(
get_us_trading_dates_holidays("trade", "last", last=5), pd.DataFrame
)
assert isinstance(
get_us_trading_dates_holidays("trade", "last", startDate="20200502"),
pd.DataFrame,
)
assert isinstance(
get_us_trading_dates_holidays(
"trade", "last", last=2, startDate="20191201"
),
pd.DataFrame,
)
assert isinstance(
get_us_trading_dates_holidays("holiday", "next"), pd.DataFrame
)