Method test_array_input

tests/transforms/test_to_tensor.py:37–41

(self, test_data, expected_shape)

class TestToTensor(unittest.TestCase):
    @parameterized.expand(TESTS)
    def test_array_input(self, test_data, expected_shape):
        result = ToTensor(dtype=torch.float32, device="cpu", wrap_sequence=True)(test_data)
        self.assertTrue(isinstance(result, torch.Tensor))
        assert_allclose(result, test_data, type_test=False)
        self.assertTupleEqual(result.shape, expected_shape)

    @parameterized.expand(TESTS_SINGLE)
    def test_single_input(self, test_data):