Skip to content

Commit e146d93

Browse files
committed
Write basic test cases
1 parent d62717b commit e146d93

File tree

1 file changed

+23
-0
lines changed

1 file changed

+23
-0
lines changed
Lines changed: 23 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,23 @@
1+
from datetime import datetime
2+
3+
from pandas_datareader import DataReader
4+
import pytest
5+
6+
7+
class TestNaver(object):
8+
@pytest.mark.parametrize(
9+
"symbol, start, end",
10+
[
11+
("005930", (2019, 10, 1), (2019, 10, 7)),
12+
("000660", (2018, 1, 1), (2018, 12, 31)),
13+
("069500", (2017, 6, 3), (2018, 9, 9)),
14+
],
15+
)
16+
def test_naver_daily_reader(self, symbol, start, end):
17+
start = datetime(*start)
18+
end = datetime(*end)
19+
reader = DataReader(symbol, "naver", start, end)
20+
21+
assert reader.shape[1] == 6
22+
assert reader["Date"].min() >= start
23+
assert reader["Date"].max() <= end

0 commit comments

Comments
 (0)