(self)
| 93 | self.assertEqual(result.shape, expected_shape) |
| 94 | |
| 95 | def test_script(self): |
| 96 | input_param, input_shape, expected_shape = TEST_CASE_SEGRESNET_VAE[0] |
| 97 | net = SegResNetVAE(**input_param) |
| 98 | test_data = torch.randn(input_shape) |
| 99 | test_script_save(net, test_data) |
| 100 | |
| 101 | |
| 102 | if __name__ == "__main__": |
nothing calls this directly
no test coverage detected