class TdxDataSourceTest(TestCase):
def setUp(self):
self.tdx = TdxDataSource(os.path.join(get_file_dir(__file__), 'data'))
def test_read_kline(self):
df = next(self.tdx.read_kline('sz002707'))
assert_that(df.columns.name, equal_to('sz002707'))
data = df.ix[20150608]
assert_that(data.amount, close_to(457520100, 100))
assert_that(data.volume, equal_to(3750037))
assert_that(data.open, equal_to(126.87))
assert_that(data.high, equal_to(130.45))
assert_that(data.low, equal_to(117.60))
assert_that(data.close, equal_to(119.64))
def test_read_minutes(self):
df = next(self.tdx.read_1_minute('sh600000'))
assert_that(df.columns.name, equal_to('sh600000'))
test_data = df.ix[20150522, 1018]
assert_that(test_data.open, equal_to(17.73))
assert_that(test_data.close, equal_to(17.74))
assert_that(test_data.high, equal_to(17.75))
assert_that(test_data.low, equal_to(17.72))
assert_that(test_data.amount, equal_to(32970880))
assert_that(test_data.volume, equal_to(1859300))
def test_read_5_minutes(self):
df = next(self.tdx.read_5_minute('sh600010'))
assert_that(df.columns.name, equal_to('sh600010'))
test_data = df.ix[20150812, 1430]
assert_that(test_data.open, equal_to(4.77))
assert_that(test_data.close, equal_to(4.78))
assert_that(test_data.high, equal_to(4.79))
assert_that(test_data.low, equal_to(4.76))
assert_that(test_data.amount, equal_to(22869504))
assert_that(test_data.volume, equal_to(4796300))
def test_read_multiple_indices(self):
ret = list(self.tdx.read_kline('sh600028', 'sz002707'))
assert_that(len(ret), equal_to(2))
def setUp(self):
self.tdx = TdxDataSource(os.path.join(get_file_dir(__file__), 'data'))
def tdx(install_root_folder):
from cn_stock_src.tdx import TdxDataSource
return TdxDataSource(install_root_folder)
