(self, input_param, img, expected_value)
| 60 | class TestMeanEnsemble(unittest.TestCase): |
| 61 | @parameterized.expand(TESTS) |
| 62 | def test_value(self, input_param, img, expected_value): |
| 63 | result = MeanEnsemble(**input_param)(img) |
| 64 | assert_allclose(result, expected_value) |
| 65 | |
| 66 | def test_cuda_value(self): |
| 67 | img = torch.stack([torch.ones(2, 2, 2, 2), torch.ones(2, 2, 2, 2) + 2]) |
nothing calls this directly
no test coverage detected