Method test_0_qlib_data

tests/test_get_data.py:35–41

(self)

        shutil.rmtree(str(DATA_DIR.resolve()))

    def test_0_qlib_data(self):
        GetData().qlib_data(
            name="qlib_data_simple", target_dir=QLIB_DIR, region="cn", interval="1d", delete_old=False, exists_skip=True
        )
        df = D.features(D.instruments("csi300"), self.FIELDS)
        self.assertListEqual(list(df.columns), self.FIELDS, "get qlib data failed")
        self.assertFalse(df.dropna().empty, "get qlib data failed")

    def test_1_csv_data(self):
        GetData().download_data(file_name="csv_data_cn.zip", target_dir=SOURCE_DIR)