(self, input_param, input_shape, expected_shape)
| 87 | class TestResNetVAE(unittest.TestCase): |
| 88 | @parameterized.expand(TEST_CASE_SEGRESNET_VAE) |
| 89 | def test_vae_shape(self, input_param, input_shape, expected_shape): |
| 90 | net = SegResNetVAE(**input_param).to(device) |
| 91 | with eval_mode(net): |
| 92 | result, _ = net(torch.randn(input_shape).to(device)) |
| 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] |
nothing calls this directly
no test coverage detected