(self)
| 93 | self.assertEqual(result.shape, expected_shape) |
| 94 | |
| 95 | def test_script(self): |
| 96 | net = BasicUNet(spatial_dims=2, in_channels=1, out_channels=3) |
| 97 | test_data = torch.randn(16, 1, 32, 32) |
| 98 | test_script_save(net, test_data) |
| 99 | |
| 100 | |
| 101 | if __name__ == "__main__": |
nothing calls this directly
no test coverage detected