360 to 180 weigths to floats + a test (#138)

* turning weights into floats and adding a test with real data

* turning weights into floats2

Author: dkazanc

httomolibgpu/misc/morph.py

@@ -83,15 +83,15 @@ def __sino_360_to_180(
     out = cp.empty((n, dy, 2 * dz - overlap), dtype=data.dtype)
 
     if rotation == "left":
-        weights = cp.linspace(0, 1.0, overlap)
+        weights = cp.linspace(0, 1.0, overlap, dtype=cp.float32)
         out[:, :, -dz + overlap :] = data[:n, :, overlap:]
         out[:, :, : dz - overlap] = data[n : 2 * n, :, overlap:][:, :, ::-1]
         out[:, :, dz - overlap : dz] = (
             weights * data[:n, :, :overlap]
             + (weights * data[n : 2 * n, :, :overlap])[:, :, ::-1]
         )
     elif rotation == "right":
-        weights = cp.linspace(1.0, 0, overlap)
+        weights = cp.linspace(1.0, 0, overlap, dtype=cp.float32)
         out[:, :, : dz - overlap] = data[:n, :, :-overlap]
         out[:, :, -dz + overlap :] = data[n : 2 * n, :, :-overlap][:, :, ::-1]
         out[:, :, dz - overlap : dz] = (

tests/test_misc/test_morph.py

@@ -7,6 +7,21 @@
 from httomolibgpu.misc.morph import sino_360_to_180, data_resampler
 
 
+
+@pytest.mark.parametrize("rotation", ["left", "right"])
+def test_sino_360_to_180_realdata(ensure_clean_memory, sino3600, rotation):
+    shape_new = (3601, 3, 2560)
+    data3d = cp.zeros(shape_new, dtype=np.float32)
+    data3d[:,0,:] = sino3600
+    data3d[:,1,:] = sino3600
+    data3d[:,2,:] = sino3600
+
+    sino_360_to_180(data3d, overlap=900, rotation=rotation)
+    
+    assert data3d.shape == shape_new
+    assert data3d.dtype == np.float32
+    assert data3d.flags.c_contiguous    
+
 @pytest.mark.parametrize(
     "overlap, rotation",
     [