fixing tests as data checker accepts only float and uint16 data

Author: dkazanc

httomolibgpu/misc/supp_func.py

@@ -167,7 +167,7 @@ def data_checker(
     Parameters
     ----------
     data : xp.ndarray
-        Input CuPy or Numpy array.
+        Input CuPy or Numpy array either float32 or uint16 data type.
     verbosity : bool
         If enabled, then the printing of the warning happens when data contains infs or nans.
     method_name : str, optional.

httomolibgpu/prep/alignment.py

@@ -88,9 +88,9 @@ def distortion_correction_proj_discorpy(
     if len(data.shape) == 2:
         data = cp.expand_dims(data, axis=0)
 
-    data = data_checker(
-        data, verbosity=True, method_name="distortion_correction_proj_discorpy"
-    )
+    # data = data_checker(
+    #     data, verbosity=True, method_name="distortion_correction_proj_discorpy"
+    # )
 
     # Get info from metadata txt file
     xcenter, ycenter, list_fact = _load_metadata_txt(metadata_path)

tests/test_prep/test_alignment.py

@@ -38,7 +38,7 @@ def test_correct_distortion(
 
     path = os.path.join(distortion_correction_path, image)
     im_host = imread(path)
-    im = cp.asarray(im_host)
+    im = cp.asarray(im_host.astype(cp.uint16))
 
     shift_xy = [0, 0]
     step_xy = [1, 1]
@@ -48,5 +48,5 @@ def test_correct_distortion(
 
     assert_allclose(np.mean(corrected_data), mean_value)
     assert np.max(corrected_data) == max_value
-    assert corrected_data.dtype == np.uint8
+    assert corrected_data.dtype == cp.uint16
     assert corrected_data.flags.c_contiguous