mirror of
https://github.com/microsoft/qlib.git
synced 2026-07-14 16:26:55 +08:00
Fix tests
This commit is contained in:
@@ -41,7 +41,7 @@
|
|||||||
" print(f\"Qlib data is not found in {provider_uri}\")\n",
|
" print(f\"Qlib data is not found in {provider_uri}\")\n",
|
||||||
" sys.path.append(str(CUR_DIR.parent.parent.joinpath(\"scripts\")))\n",
|
" sys.path.append(str(CUR_DIR.parent.parent.joinpath(\"scripts\")))\n",
|
||||||
" from get_data import GetData\n",
|
" from get_data import GetData\n",
|
||||||
" GetData().qlib_data_cn(provider_uri)\n",
|
" GetData().qlib_data_cn(target_dir=provider_uri)\n",
|
||||||
"qlib.init(provider_uri=provider_uri, region=REG_CN)"
|
"qlib.init(provider_uri=provider_uri, region=REG_CN)"
|
||||||
]
|
]
|
||||||
},
|
},
|
||||||
|
|||||||
@@ -26,7 +26,7 @@ if __name__ == "__main__":
|
|||||||
sys.path.append(str(Path(__file__).resolve().parent.parent.joinpath("scripts")))
|
sys.path.append(str(Path(__file__).resolve().parent.parent.joinpath("scripts")))
|
||||||
from get_data import GetData
|
from get_data import GetData
|
||||||
|
|
||||||
GetData().qlib_data_cn(provider_uri)
|
GetData().qlib_data_cn(target_dir=provider_uri)
|
||||||
|
|
||||||
qlib.init(provider_uri=provider_uri, region=REG_CN)
|
qlib.init(provider_uri=provider_uri, region=REG_CN)
|
||||||
|
|
||||||
|
|||||||
@@ -37,7 +37,7 @@
|
|||||||
" print(f\"Qlib data is not found in {provider_uri}\")\n",
|
" print(f\"Qlib data is not found in {provider_uri}\")\n",
|
||||||
" sys.path.append(str(Path.cwd().parent.joinpath(\"scripts\")))\n",
|
" sys.path.append(str(Path.cwd().parent.joinpath(\"scripts\")))\n",
|
||||||
" from get_data import GetData\n",
|
" from get_data import GetData\n",
|
||||||
" GetData().qlib_data_cn(provider_uri)\n",
|
" GetData().qlib_data_cn(target_dir=provider_uri)\n",
|
||||||
"qlib.init(provider_uri=provider_uri, region=REG_CN)"
|
"qlib.init(provider_uri=provider_uri, region=REG_CN)"
|
||||||
]
|
]
|
||||||
},
|
},
|
||||||
|
|||||||
@@ -37,7 +37,7 @@ class TestGetData(unittest.TestCase):
|
|||||||
|
|
||||||
def test_0_qlib_data(self):
|
def test_0_qlib_data(self):
|
||||||
|
|
||||||
GetData().qlib_data_cn(QLIB_DIR)
|
GetData().qlib_data_cn(name="qlib_data_cn_simple", target_dir=QLIB_DIR)
|
||||||
df = D.features(D.instruments("csi300"), self.FIELDS)
|
df = D.features(D.instruments("csi300"), self.FIELDS)
|
||||||
self.assertListEqual(list(df.columns), self.FIELDS, "get qlib data failed")
|
self.assertListEqual(list(df.columns), self.FIELDS, "get qlib data failed")
|
||||||
self.assertFalse(df.dropna().empty, "get qlib data failed")
|
self.assertFalse(df.dropna().empty, "get qlib data failed")
|
||||||
|
|||||||
Reference in New Issue
Block a user