| 156 | class TestHoverNet(unittest.TestCase): |
| 157 | @parameterized.expand(CASES) |
| 158 | def test_shape(self, input_param, input_shape, expected_shapes): |
| 159 | input_param["decoder_padding"] = False |
| 160 | net = HoVerNet(**input_param).to(device) |
| 161 | with eval_mode(net): |
| 162 | result = net.forward(torch.randn(input_shape).to(device)) |
| 163 | for item in result: |
| 164 | self.assertEqual(result[item].shape, expected_shapes[item]) |
| 165 | |
| 166 | @parameterized.expand(CASES) |
| 167 | def test_decoder_padding_shape(self, input_param, input_shape, expected_shapes): |