| 12 | return "".join(str(obj).split()) |
| 13 | |
| 14 | def test_handler_df(self): |
| 15 | df = D.features(["sh600519"], start_time="20190101", end_time="20190201", fields=["$close"]) |
| 16 | dh = DataHandlerLP.from_df(df) |
| 17 | print(dh.fetch()) |
| 18 | self.assertTrue(dh._data.equals(df)) |
| 19 | self.assertTrue(dh._infer is dh._data) |
| 20 | self.assertTrue(dh._learn is dh._data) |
| 21 | self.assertTrue(dh.data_loader._data is dh._data) |
| 22 | fname = "_handler_test.pkl" |
| 23 | dh.to_pickle(fname, dump_all=True) |
| 24 | |
| 25 | with open(fname, "rb") as f: |
| 26 | dh_d = pickle.load(f) |
| 27 | |
| 28 | self.assertTrue(dh_d._data.equals(df)) |
| 29 | self.assertTrue(dh_d._infer is dh_d._data) |
| 30 | self.assertTrue(dh_d._learn is dh_d._data) |
| 31 | # Data loader will no longer be useful |
| 32 | self.assertTrue("_data" not in dh_d.data_loader.__dict__.keys()) |
| 33 | os.remove(fname) |
| 34 | |
| 35 | |
| 36 | if __name__ == "__main__": |