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')
#!/usr/bin/env python # -*- coding: utf-8 -*- from datasource.test_csv import * from test_engine import * from trading.test_future import * from trading.test_stock import * from quantdigger import ConfigUtil if __name__ == '__main__': # 默认为csv assert(ConfigUtil.get('source') == 'csv') unittest.main() assert(ConfigUtil.get('source') == 'csv') # 这里的代码不会被运行 # code
#!/usr/bin/env python # -*- coding: utf-8 -*- from datasource.test_csv import * from test_engine import * from test_future import * from test_stock import * from quantdigger import ConfigUtil if __name__ == '__main__': # 默认为csv assert (ConfigUtil.get('source') == 'csv') unittest.main() assert (ConfigUtil.get('source') == 'csv') # 这里的代码不会被运行 # code