def test_simple(self):
x, y, n = 2, 3, 10
peak_array = np.arange(y * x * n * 2).reshape((y, x, n, 2))
inlier_array = np.ones((y, x, n), dtype=bool)
inlier_parray0, outlier_parray0 = ret._get_inlier_outlier_peak_arrays(
peak_array, inlier_array)
inlier_parray1, outlier_parray1 = ret._get_inlier_outlier_peak_arrays(
peak_array, ~inlier_array)
for iy, ix in np.ndindex(peak_array.shape[:2]):
assert len(inlier_parray0[iy, ix]) == n
assert len(outlier_parray0[iy, ix]) == 0
assert len(inlier_parray1[iy, ix]) == 0
assert len(outlier_parray1[iy, ix]) == n
def test_some_true_some_false(self):
x, y, n = 2, 3, 10
peak_array = np.arange(y * x * n * 2).reshape((y, x, n, 2))
inlier_array = np.ones((y, x, n), dtype=bool)
inlier_array[:, :, 4:] = False
inlier_parray, outlier_parray = ret._get_inlier_outlier_peak_arrays(
peak_array, inlier_array)
for iy, ix in np.ndindex(peak_array.shape[:2]):
assert len(inlier_parray[iy, ix]) == 4
assert len(outlier_parray[iy, ix]) == 6
assert (peak_array[iy, ix][:4] == inlier_parray[iy, ix]).all()
assert (peak_array[iy, ix][4:] == outlier_parray[iy, ix]).all()
def test_inlier_none(self):
x, y, n = 2, 3, 10
peak_array = np.arange(y * x * n * 2).reshape((y, x, n, 2))
inlier_array = np.empty((y, x), dtype=object)
for iy, ix in np.ndindex(inlier_array.shape):
inlier_array[iy, ix] = None
inlier_parray, outlier_parray = ret._get_inlier_outlier_peak_arrays(
peak_array, inlier_array)
for iy, ix in np.ndindex(inlier_array.shape):
assert inlier_parray[iy, ix] is None
assert len(outlier_parray[iy, ix]) == n
