def test_visu_slope_corrector_slice_wise_slope_5d_fail():
in_data = np.random.normal(5, 10, [2, 3, 4, 5, 6])
sl = np.random.normal(5, 10, 7)
with assert_raises(IOError):
visu_slope_corrector(in_data, sl)
def test_visu_slope_corrector_int_float_slope():
in_data = np.random.normal(5, 10, [3, 4, 5])
sl1 = 5
sl2 = 5.3
out_data1 = visu_slope_corrector(in_data, sl1, dtype=np.float64)
out_data2 = visu_slope_corrector(in_data, sl2, dtype=np.float64)
assert_equal(out_data1.dtype, np.float64)
assert_equal(out_data2.dtype, np.float64)
assert_array_equal(out_data1, sl1 * in_data)
assert_array_equal(out_data2, sl2 * in_data)
def test_visu_slope_corrector_slice_wise_slope_5d():
in_data = np.random.normal(5, 10, [2, 3, 4, 5, 6])
sl = np.random.normal(5, 10, 5)
out_data = visu_slope_corrector(in_data, sl, dtype=np.float64)
for t in range(6):
for k in range(5):
assert_array_equal(out_data[..., k, t], in_data[..., k, t] * sl[k])
def test_visu_slope_corrector_slice_wise_slope_3d():
in_data = np.random.normal(5, 10, [4, 5, 3])
sl = np.random.normal(5, 10, 3)
out_data = visu_slope_corrector(in_data, sl, dtype=np.float64)
for k in range(3):
assert_array_equal(out_data[..., k], in_data[..., k] * sl[k])