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)