Add zenodo test

Author: namannimmo10

zenodo-tests/test_prep/test_stripe.py

@@ -0,0 +1,33 @@
+import time
+import cupy as cp
+import numpy as np
+import pytest
+
+from numpy.testing import assert_allclose
+from httomolibgpu.prep.stripe import remove_all_stripe
+from httomolibgpu.prep.normalize import normalize
+from conftest import force_clean_gpu_memory
+
+
+@pytest.mark.parametrize(
+    "dataset_fixture, exp_sum, exp_mean, exp_mean_sum, exp_shape",
+    [
+        ("i13_dataset1", 32071604.0, 0.208765, 1252.7957, (6001, 10, 2560)),
+        ("i13_dataset3", 27234604.0, 0.59093, 3546.1704, (6001, 3, 2560)),
+    ],
+    ids=["dataset_one", "dataset_three"],
+)
+def test_remove_all_stripe_with_i13_data(
+    request, dataset_fixture, exp_sum, exp_mean, exp_mean_sum, exp_shape
+):
+    dataset = request.getfixturevalue(dataset_fixture)
+    data_normalised = normalize(dataset[0], dataset[2], dataset[3], minus_log=True)
+    output = remove_all_stripe(cp.copy(data_normalised)).get()
+
+    assert_allclose(np.sum(output), exp_sum, rtol=1e-07, atol=1e-6)
+    assert_allclose(np.mean(output), exp_mean, rtol=1e-07, atol=1e-6)
+    assert_allclose(np.mean(output, axis=(1, 2)).sum(), exp_mean_sum, rtol=1e-06)
+    force_clean_gpu_memory()
+    assert output.shape == exp_shape
+    assert output.dtype == np.float32
+    assert output.flags.c_contiguous

zenodo-tests/test_recon/test_algorithm.py

@@ -38,7 +38,7 @@ def test_reconstruct_FBP_i12_dataset1(i12_dataset1):
     )
     assert recon_data.flags.c_contiguous
     recon_data = recon_data.get()
-    assert_allclose(np.sum(recon_data), 46569.395, rtol=1e-07, atol=1e-6)
+    assert_allclose(np.sum(recon_data), 46569.402, rtol=1e-07, atol=1e-6)
     assert recon_data.dtype == np.float32
     assert recon_data.shape == (2560, 50, 2560)