liuyu/qlib
1
0
Fork 0
mirror of https://github.com/microsoft/qlib.git synced 2026-06-30 17:41:18 +08:00
Code Activity

update tests

Browse Source
This commit is contained in:
bxdd
2021-01-26 08:47:07 +00:00
parent 3f9f295a87
commit 2016ebbbb2
2 changed files with 6 additions and 8 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
qlib/tests/__init__.py
View File

@@ -7,6 +7,9 @@ from ..config import REG_CN
class TestAutoData(unittest.TestCase):
_setup_kwargs = {}
@classmethod
def setUpClass(cls) -> None:
# use default data
@@ -17,4 +20,4 @@ class TestAutoData(unittest.TestCase):
GetData().qlib_data(
name="qlib_data_simple", region="cn", version="latest", interval="1d", target_dir=provider_uri
)
init(provider_uri=provider_uri, region=REG_CN)
init(provider_uri=provider_uri, region=REG_CN, **cls._setup_kwargs)

9
tests/test_register_ops.py
View File

@@ -62,13 +62,8 @@ class Distance(PairOperator):
class TestRegiterCustomOps(TestAutoData):
@classmethod
def setUpClass(cls) -> None:
# use default data
provider_uri = "~/.qlib/qlib_data/cn_data_simple_1" # target_dir
if not exists_qlib_data(provider_uri):
print(f"Qlib data is not found in {provider_uri}")
GetData().qlib_data(name="qlib_data_simple", region="cn", interval="1d", target_dir=provider_uri)
qlib.init(provider_uri=provider_uri, custom_ops=[Diff, Distance], region=REG_CN)
cls._setup_kwargs.update({"custom_ops":[Diff, Distance]})
super().setUpClass()
def test_regiter_custom_ops(self):
instruments = ["SH600000"]