From 2016ebbbb2b3cb911b517dc842b7793d5cdad477 Mon Sep 17 00:00:00 2001
From: bxdd <bxd98@126.com>
Date: Tue, 26 Jan 2021 08:47:07 +0000
Subject: [PATCH] update tests

---
 qlib/tests/__init__.py     | 5 ++++-
 tests/test_register_ops.py | 9 ++-------
 2 files changed, 6 insertions(+), 8 deletions(-)

diff --git a/qlib/tests/__init__.py b/qlib/tests/__init__.py
index a1b33a2a2..bdc6297da 100644
--- a/qlib/tests/__init__.py
+++ b/qlib/tests/__init__.py
@@ -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)
diff --git a/tests/test_register_ops.py b/tests/test_register_ops.py
index d497d0b89..ec2ec3d93 100644
--- a/tests/test_register_ops.py
+++ b/tests/test_register_ops.py
@@ -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"]