(self, src, expected_keys, create_dir=True, atomic=True, func=None, kwargs=None)
| 59 | ] |
| 60 | ) |
| 61 | def test_file(self, src, expected_keys, create_dir=True, atomic=True, func=None, kwargs=None): |
| 62 | with tempfile.TemporaryDirectory() as tempdir: |
| 63 | path = os.path.join(tempdir, "test_ckpt.pt") |
| 64 | if kwargs is None: |
| 65 | kwargs = {} |
| 66 | save_state(src=src, path=path, create_dir=create_dir, atomic=atomic, func=func, **kwargs) |
| 67 | ckpt = dict(torch.load(path, weights_only=True)) |
| 68 | for k in ckpt.keys(): |
| 69 | self.assertIn(k, expected_keys) |
| 70 | |
| 71 | |
| 72 | if __name__ == "__main__": |
nothing calls this directly
no test coverage detected