mirror of
https://github.com/microsoft/qlib.git
synced 2026-07-02 10:31:00 +08:00
28 lines
763 B
Python
28 lines
763 B
Python
import sys
|
|
import unittest
|
|
from ..utils import exists_qlib_data
|
|
from .data import GetData
|
|
from .. import init
|
|
from ..config import REG_CN
|
|
|
|
|
|
class TestAutoData(unittest.TestCase):
|
|
|
|
_setup_kwargs = {}
|
|
|
|
@classmethod
|
|
def setUpClass(cls) -> None:
|
|
# use default data
|
|
provider_uri = "~/.qlib/qlib_data/cn_data_simple" # 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,
|
|
delete_old=False,
|
|
)
|
|
init(provider_uri=provider_uri, region=REG_CN, **cls._setup_kwargs)
|