(self, input_param, input_shape, expected_shape)
| 65 | class TestSegResNetDS(unittest.TestCase): |
| 66 | @parameterized.expand(TEST_CASE_SEGRESNET_DS) |
| 67 | def test_shape(self, input_param, input_shape, expected_shape): |
| 68 | net = SegResNetDS(**input_param).to(device) |
| 69 | with eval_mode(net): |
| 70 | result = net(torch.randn(input_shape).to(device)) |
| 71 | self.assertEqual(result.shape, expected_shape, msg=str(input_param)) |
| 72 | |
| 73 | @parameterized.expand(TEST_CASE_SEGRESNET_DS) |
| 74 | def test_shape_ds2(self, input_param, input_shape, expected_shape): |
nothing calls this directly
no test coverage detected