add test for apply_corr

Author: yucongalicechen

src/diffpy/labpdfproc/tests/test_functions.py

@@ -1,7 +1,7 @@
 import numpy as np
 import pytest
 
-from diffpy.labpdfproc.functions import Gridded_circle, compute_cve
+from diffpy.labpdfproc.functions import Gridded_circle, apply_corr, compute_cve
 from diffpy.utils.scattering_objects.diffraction_objects import Diffraction_object
 
 params1 = [
@@ -83,3 +83,20 @@ def test_compute_cve(mocker):
         scat_quantity="cve",
     )
     assert actual_abdo == expected_abdo
+
+
+def test_apply_corr(mocker):
+    xarray, yarray = np.array([90, 90.1, 90.2]), np.array([2, 2, 2])
+    expected_cve = np.array([0.5, 0.5, 0.5])
+    mocker.patch("diffpy.labpdfproc.functions.TTH_GRID", xarray)
+    mocker.patch("numpy.interp", return_value=expected_cve)
+    input_pattern = _instantiate_test_do(xarray, yarray)
+    absorption_correction = _instantiate_test_do(
+        xarray,
+        expected_cve,
+        name="absorption correction, cve, for test",
+        scat_quantity="cve",
+    )
+    actual_corr = apply_corr(input_pattern, absorption_correction)
+    expected_corr = _instantiate_test_do(xarray, np.array([1, 1, 1]))
+    assert actual_corr == expected_corr