Merge branch 'main' into add-to-video

Author: fepegar

docs/source/transforms/preprocessing.rst

@@ -46,6 +46,13 @@ Intensity
     :show-inheritance:
 
 
+:class:`To`
+~~~~~~~~~~~
+
+.. autoclass:: To
+    :show-inheritance:
+
+
 .. currentmodule:: torchio.transforms.preprocessing.intensity
 
 
@@ -56,13 +63,6 @@ Intensity
     :show-inheritance:
 
 
-:class:`To`
-~~~~~~~~~~~
-
-.. autoclass:: To
-    :show-inheritance:
-
-
 Spatial
 -------
 

src/torchio/transforms/preprocessing/intensity/to.py

@@ -12,6 +12,8 @@
 class To(IntensityTransform):
     """Convert the image tensor data type and/or device.
 
+    This transform is a thin wrapper around :func:`torch.Tensor.to`.
+
     Args:
         target: First argument to :func:`torch.Tensor.to`.
         to_kwargs: Additional keyword arguments to pass to :func:`torch.Tensor.to`.
@@ -25,7 +27,6 @@ class To(IntensityTransform):
         >>> ct_rescaled = rescale(ct_clamped)
         >>> to_uint8 = tio.To(torch.uint8)
         >>> ct_uint8 = to_uint8(ct_rescaled)
-
     """
 
     def __init__(

tests/transforms/preprocessing/test_to.py

@@ -1,2 +1,17 @@
-def test_todo():
-    assert False
+import torch
+
+import torchio as tio
+
+from ...utils import TorchioTestCase
+
+
+class TestTo(TorchioTestCase):
+    """Tests for :class:`tio.To` class."""
+
+    def test_to(self):
+        transform = tio.To(torch.int)
+        tensor = 10 * torch.rand(2, 3, 4, 5)
+        image = tio.ScalarImage(tensor=tensor)
+        transformed = transform(image)
+        assert image.data.dtype == torch.float32
+        assert transformed.data.dtype == torch.int