()
| 136 | |
| 137 | |
| 138 | def test_serialize_tensors(): |
| 139 | # create a temporary file and immediately release it so DGL can open it. |
| 140 | f = tempfile.NamedTemporaryFile(delete=False) |
| 141 | path = f.name |
| 142 | f.close() |
| 143 | |
| 144 | tensor_dict = { |
| 145 | "a": F.tensor([1, 3, -1, 0], dtype=F.int64), |
| 146 | "1@1": F.tensor([1.5, 2], dtype=F.float32), |
| 147 | } |
| 148 | |
| 149 | save_tensors(path, tensor_dict) |
| 150 | |
| 151 | load_tensor_dict = load_tensors(path) |
| 152 | |
| 153 | for key in tensor_dict: |
| 154 | assert key in load_tensor_dict |
| 155 | assert np.array_equal( |
| 156 | F.asnumpy(load_tensor_dict[key]), F.asnumpy(tensor_dict[key]) |
| 157 | ) |
| 158 | |
| 159 | load_nd_dict = load_tensors(path, return_dgl_ndarray=True) |
| 160 | |
| 161 | for key in tensor_dict: |
| 162 | assert key in load_nd_dict |
| 163 | assert isinstance(load_nd_dict[key], nd.NDArray) |
| 164 | assert np.array_equal( |
| 165 | load_nd_dict[key].asnumpy(), F.asnumpy(tensor_dict[key]) |
| 166 | ) |
| 167 | |
| 168 | os.unlink(path) |
| 169 | |
| 170 | |
| 171 | def test_serialize_empty_dict(): |
nothing calls this directly
no test coverage detected