(self, input, expected_value, wrap_array=False)
| 39 | class TestEnsureTuple(unittest.TestCase): |
| 40 | @parameterized.expand(TESTS) |
| 41 | def test_value(self, input, expected_value, wrap_array=False): |
| 42 | result = ensure_tuple(input, wrap_array) |
| 43 | |
| 44 | self.assertTrue(isinstance(result, tuple)) |
| 45 | if isinstance(input, (np.ndarray, torch.Tensor)): |
| 46 | for i, j in zip(result, expected_value): |
| 47 | assert_allclose(i, j) |
| 48 | else: |
| 49 | self.assertTupleEqual(result, expected_value) |
| 50 | |
| 51 | |
| 52 | if __name__ == "__main__": |
nothing calls this directly
no test coverage detected