(self)
| 81 | |
| 82 | @unittest.skipIf(not config.USE_META_DICT, "not using meta dict") |
| 83 | def test_meta_dict(self): |
| 84 | xform = Flipd("image", [0, 1]) |
| 85 | res = xform({"image": torch.zeros(1, 3, 4)}) |
| 86 | self.assertEqual(res["image"].applied_operations, res["image_transforms"]) |
| 87 | |
| 88 | |
| 89 | if __name__ == "__main__": |