(self, input_param, input_shape, expected_shape)
| 50 | class TestHighResNet(DistTestCase): |
| 51 | @parameterized.expand([TEST_CASE_1, TEST_CASE_2, TEST_CASE_3, TEST_CASE_4]) |
| 52 | def test_shape(self, input_param, input_shape, expected_shape): |
| 53 | net = HighResNet(**input_param).to(device) |
| 54 | with eval_mode(net): |
| 55 | result = net.forward(torch.randn(input_shape).to(device)) |
| 56 | self.assertEqual(result.shape, expected_shape) |
| 57 | |
| 58 | @TimedCall(seconds=800, force_quit=True) |
| 59 | def test_script(self): |
nothing calls this directly
no test coverage detected