expose test cases updated per review comments

rapidsai · Sep 28, 2021 · bdfd569 · bdfd569
1 parent a94055e
commit bdfd569
Showing 1 changed file with 7 additions and 11 deletions.
diff --git a/python/cucim/src/cucim/core/operations/expose/tests/test_expose.py b/python/cucim/src/cucim/core/operations/expose/tests/test_expose.py
@@ -1,20 +1,16 @@
-from cucim.core.operations.expose.transform.expose import (
-    color_jitter,
-    image_flip,
-    image_rotate_90,
-    scale_intensity,
-    zoom,
-    rand_zoom,
-    rand_image_flip,
-    rand_image_rotate_90,
-)
+from cucim.core.operations.expose.transform import (color_jitter, image_flip,
+                                                    image_rotate_90,
+                                                    rand_image_flip,
+                                                    rand_image_rotate_90,
+                                                    rand_zoom,
+                                                    scale_intensity_range, zoom)
 
 
 def test_exposed_transforms():
     assert color_jitter is not None
     assert image_flip is not None
     assert image_rotate_90 is not None
-    assert scale_intensity is not None
+    assert scale_intensity_range is not None
     assert zoom is not None
     assert rand_zoom is not None
     assert rand_image_flip is not None