(self, input_param, input_shape, expected_shape)
| 65 | class TestMilModel(unittest.TestCase): |
| 66 | @parameterized.expand(TEST_CASE_MILMODEL) |
| 67 | def test_shape(self, input_param, input_shape, expected_shape): |
| 68 | with skip_if_downloading_fails(): |
| 69 | net = MILModel(**input_param).to(device) |
| 70 | with eval_mode(net): |
| 71 | result = net(torch.randn(input_shape, dtype=torch.float).to(device)) |
| 72 | self.assertEqual(result.shape, expected_shape) |
| 73 | |
| 74 | def test_ill_args(self): |
| 75 | with self.assertRaises(ValueError): |
nothing calls this directly
no test coverage detected