def test_validate_gt_mask_valid(self) -> None:
        """Test validation of a valid ground truth mask."""
        mask = torch.randint(0, 2, (1, 224, 224))
        validated_mask = self.validator.validate_gt_mask(mask)
        assert isinstance(validated_mask, Mask)
        assert validated_mask.shape == (224, 224)
        assert validated_mask.dtype == torch.bool
Test validation of a valid ground truth mask.
tests/unit/data/validators/torch/test_image.py