Source code for crikit.cri.tests.test_merge_nrbs
import numpy as np
from crikit.cri.merge_nrbs import MergeNRBs
[docs]def test_basic():
x = np.arange(0,1000)
nrb_left = np.exp(-(x-500)**2/(100**2))
nrb_right = np.exp(-(x-700)**2/(120**2))
pix = 625
# Left scale
merge = MergeNRBs(nrb_left, nrb_right, pix, left_side_scale=True)
out_scaled_left = merge.calculate()
assert nrb_right[pix] == out_scaled_left[pix]
# Right scale
merge = MergeNRBs(nrb_left, nrb_right, pix, left_side_scale=False)
out_scaled_right = merge.calculate()
assert nrb_left[pix] == out_scaled_right[pix]
merge = MergeNRBs(nrb_left, nrb_right, pix, left_side_scale=None)
out_scaled_none = merge.calculate()
assert nrb_right[pix] == out_scaled_none[pix]
assert nrb_left[pix-1] == out_scaled_none[pix-1]