def test_local_data2(self):
old_source = ConfigUtil.get("source")
if old_source == "csv":
return
logger.info("***** 数据测试开始 *****")
ConfigUtil.set(source="sqlite")
data_manager = DataManager()
target = data_manager.get_bars("BB.TEST-1.Minute", _DT_START, _DT_END).data
fname = os.path.join(os.getcwd(), "data", "1MINUTE", "TEST", "CC.csv")
source = pd.read_csv(fname, parse_dates="datetime", index_col="datetime")
self.assertFalse(source.equals(target), "本地数据接口负测试失败！")
fname = os.path.join(os.getcwd(), "data", "1MINUTE", "TEST", "BB.csv")
source = pd.read_csv(fname, parse_dates="datetime", index_col="datetime")
self.assertTrue(source.equals(target), "本地数据接口正测试失败！")
logger.info("-- 本地数据接口测试成功 --")
logger.info("***** 数据测试结束 *****\n")
ConfigUtil.set(source=old_source)
def test_csv_source(self):
source_bak = ConfigUtil.get('source')
logger.info('***** 数据测试开始 *****')
ConfigUtil.set(source='csv')
data_manager = DataManager()
target = data_manager.get_bars(
'BB.TEST-1.Minute', _DT_START, _DT_END).data
fname = os.path.join(os.getcwd(), 'data', '1MINUTE', 'TEST', 'CC.csv')
source = pd.read_csv(
fname, parse_dates='datetime', index_col='datetime')
self.assertFalse(source.equals(target), '本地数据接口负测试失败！')
fname = os.path.join(os.getcwd(), 'data', '1MINUTE', 'TEST', 'BB.csv')
source = pd.read_csv(
fname, parse_dates='datetime', index_col='datetime')
self.assertTrue(source.equals(target), '本地数据接口正测试失败！')
logger.info('-- 本地数据接口测试成功 --')
ConfigUtil.set(source=source_bak)
logger.info('***** 数据测试结束 *****\n')
def test_csv_source(self):
source_bak = ConfigUtil.get('source')
logger.info('***** 数据测试开始 *****')
ConfigUtil.set(source='csv')
data_manager = DataManager()
target = data_manager.get_bars('BB.TEST-1.Minute', _DT_START,
_DT_END).data
fname = os.path.join(os.getcwd(), 'data', '1MINUTE', 'TEST', 'CC.csv')
source = pd.read_csv(fname,
parse_dates='datetime',
index_col='datetime')
self.assertFalse(source.equals(target), '本地数据接口负测试失败！')
fname = os.path.join(os.getcwd(), 'data', '1MINUTE', 'TEST', 'BB.csv')
source = pd.read_csv(fname,
parse_dates='datetime',
index_col='datetime')
self.assertTrue(source.equals(target), '本地数据接口正测试失败！')
logger.info('-- 本地数据接口测试成功 --')
ConfigUtil.set(source=source_bak)
logger.info('***** 数据测试结束 *****\n')
