(self, input_param, img, expected_value)
| 73 | class TestVoteEnsemble(unittest.TestCase): |
| 74 | @parameterized.expand(TESTS) |
| 75 | def test_value(self, input_param, img, expected_value): |
| 76 | result = VoteEnsemble(**input_param)(img) |
| 77 | if isinstance(img, torch.Tensor): |
| 78 | self.assertIsInstance(result, torch.Tensor) |
| 79 | self.assertEqual(result.device, img.device) |
| 80 | assert_allclose(result, expected_value) |
| 81 | |
| 82 | |
| 83 | if __name__ == "__main__": |
nothing calls this directly
no test coverage detected