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