zenodo stripe removal test with synthetic data

Author: dkazanc

.scripts/download_zenodo.py

@@ -18,15 +18,15 @@ def calculate_md5(filename):
 
 def download_zenodo_files(output_dir: Path):
     """
-    Download all files from Zenodo record 14938787 and verify their checksums.
+    Download all files from Zenodo record 14979785 and verify their checksums.
 
     Args:
         output_dir: Directory where files should be downloaded
     """
     try:
-        print("Fetching files from Zenodo record 14938787...")
+        print("Fetching files from Zenodo record 14979785...")
         with urllib.request.urlopen(
-            "https://zenodo.org/api/records/14938787"
+            "https://zenodo.org/api/records/14979785"
         ) as response:
             data = json.loads(response.read())
 

zenodo-tests/conftest.py

@@ -185,6 +185,20 @@ def geant4_dataset1(geant4_dataset1_file):
     )
 
 
+@pytest.fixture(scope="session")
+def synth_tomophantom1_file(test_data_path):
+    in_file = os.path.join(test_data_path, "synth_tomophantom1.npz")
+    return np.load(in_file)
+
+
+@pytest.fixture
+def synth_tomophantom1_dataset(synth_tomophantom1_file):
+    return (
+        cp.asarray(cp.swapaxes(synth_tomophantom1_file["projdata"], 0, 1)),
+        synth_tomophantom1_file["angles"],
+    )
+
+
 @pytest.fixture
 def ensure_clean_memory():
     gc.collect()

zenodo-tests/test_prep/test_stripe.py

@@ -134,6 +134,42 @@ def test_remove_all_stripe_i12_dataset4(
     assert output.flags.c_contiguous
 
 
+@pytest.mark.parametrize(
+    "dataset_fixture, snr_val, la_size_val, sm_size_val, norm_res_expected",
+    [
+        ("synth_tomophantom1_dataset", 1.0, 61, 21, 53435.61),
+        ("synth_tomophantom1_dataset", 0.1, 61, 21, 67917.71),
+        ("synth_tomophantom1_dataset", 0.001, 61, 21, 70015.51),
+    ],
+    ids=["snr_1", "snr_2", "snr_3"],
+)
+def test_remove_all_stripe_synth_tomophantom1_dataset(
+    request, dataset_fixture, snr_val, la_size_val, sm_size_val, norm_res_expected
+):
+    dataset = request.getfixturevalue(dataset_fixture)
+    force_clean_gpu_memory()
+
+    output = remove_all_stripe(
+        cp.copy(dataset[0]),
+        snr=snr_val,
+        la_size=la_size_val,
+        sm_size=sm_size_val,
+        dim=1,
+    )
+    np.savez(
+        "/home/algol/Documents/DEV/httomolibgpu/zenodo-tests/large_data_archive/stripe_res2.npz",
+        data=output.get(),
+    )
+
+    residual_calc = dataset[0] - output
+    norm_res = cp.linalg.norm(residual_calc.flatten())
+
+    assert isclose(norm_res, norm_res_expected, abs_tol=10**-2)
+    assert not np.isnan(output).any(), "Output contains NaN values"
+    assert output.dtype == np.float32
+    assert output.flags.c_contiguous
+
+
 @pytest.mark.parametrize(
     "dataset_fixture, nvalue_val, vvalue_val, norm_res_expected",
     [