Add unit tests for RAW tiled TIFF image

rapidsai · Sep 28, 2021 · 537c76e · 537c76e
1 parent 646c16b
commit 537c76e
Show file tree

Hide file tree

Showing 2 changed files with 35 additions and 4 deletions.
diff --git a/python/cucim/tests/fixtures/testimage.py b/python/cucim/tests/fixtures/testimage.py
@@ -46,9 +46,19 @@ def testimg_tiff_stripe_32x24_16_deflate(tmpdir_factory):
     shutil.rmtree(dataset_path)
 
 
+@pytest.fixture(scope='session')
+def testimg_tiff_stripe_32x24_16_raw(tmpdir_factory):
+    dataset_path, image_path = gen_image(
+        tmpdir_factory, 'tiff::stripe:32x24:16:raw')
+    yield image_path
+    # Clean up fake dataset folder
+    shutil.rmtree(dataset_path)
+
+
 @pytest.fixture(scope='session', params=[
     lazy_fixture('testimg_tiff_stripe_32x24_16_jpeg'),
-    lazy_fixture('testimg_tiff_stripe_32x24_16_deflate')
+    lazy_fixture('testimg_tiff_stripe_32x24_16_deflate'),
+    lazy_fixture('testimg_tiff_stripe_32x24_16_raw')
 ])
 def testimg_tiff_stripe_32x24_16(request):
     return request.param
@@ -74,9 +84,19 @@ def testimg_tiff_stripe_4096x4096_256_deflate(tmpdir_factory):
     shutil.rmtree(dataset_path)
 
 
+@pytest.fixture(scope='session')
+def testimg_tiff_stripe_4096x4096_256_raw(tmpdir_factory):
+    dataset_path, image_path = gen_image(
+        tmpdir_factory, 'tiff::stripe:4096x4096:256:raw')
+    yield image_path
+    # Clean up fake dataset folder
+    shutil.rmtree(dataset_path)
+
+
 @pytest.fixture(scope='session', params=[
     lazy_fixture('testimg_tiff_stripe_4096x4096_256_jpeg'),
-    lazy_fixture('testimg_tiff_stripe_4096x4096_256_deflate')
+    lazy_fixture('testimg_tiff_stripe_4096x4096_256_deflate'),
+    lazy_fixture('testimg_tiff_stripe_4096x4096_256_raw')
 ])
 def testimg_tiff_stripe_4096x4096_256(request):
     return request.param
@@ -101,9 +121,19 @@ def testimg_tiff_stripe_100000x100000_256_deflate(tmpdir_factory):
     shutil.rmtree(dataset_path)
 
 
+@pytest.fixture(scope='session')
+def testimg_tiff_stripe_100000x100000_256_raw(tmpdir_factory):
+    dataset_path, image_path = gen_image(
+        tmpdir_factory, 'tiff::stripe:100000x100000:256:raw')
+    yield image_path
+    # Clean up fake dataset folder
+    shutil.rmtree(dataset_path)
+
+
 @pytest.fixture(scope='session', params=[
     lazy_fixture('testimg_tiff_stripe_100000x100000_256_jpeg'),
-    lazy_fixture('testimg_tiff_stripe_100000x100000_256_deflate')
+    lazy_fixture('testimg_tiff_stripe_100000x100000_256_deflate'),
+    lazy_fixture('testimg_tiff_stripe_100000x100000_256_raw')
 ])
 def testimg_tiff_stripe_100000x100000_256(request):
     return request.param
diff --git a/python/cucim/tests/util/gen_tiff.py b/python/cucim/tests/util/gen_tiff.py
@@ -21,7 +21,8 @@
 from tifffile import TiffWriter
 
 COMPRESSION_MAP = {'jpeg': ('jpeg', 95),
-                   'deflate': 'deflate'}
+                   'deflate': 'deflate',
+                   'raw': None}
 
 
 class TiffGenerator: