def test_single_pixel_spot(self, shifts_expected, sigma): z = np.zeros((50, 50)) z[28, 24] = 1 z = gaussian_filter(z, sigma=sigma, truncate=3) shifts = find_beam_offset_cross_correlation(z, 1, 6) assert np.allclose(shifts, shifts_expected, atol=0.2)
def test_broader_starting_square_spot(self, shifts_expected): z = np.zeros((50, 50)) z[28:31, 24:27] = 1 z = gaussian_filter(z, sigma=2, truncate=3) shifts = find_beam_offset_cross_correlation(z, 1, 4) assert np.allclose(shifts, shifts_expected, atol=0.2)
def test_perfectly_centered_spot(self, shifts_expected): z = np.zeros((50, 50)) z[24:26, 24:26] = 1 z = gaussian_filter(z, sigma=2, truncate=3) shifts = find_beam_offset_cross_correlation(z, 1, 4) assert np.allclose(shifts, shifts_expected, atol=0.2)