def test_xy_errors_in_conventional_xc_method_as_per_issue_490():
""" This was the MWE example code for the issue """
dp = get_simulated_disc(100, 20)
# translate y by +4
shifted = np.pad(dp, ((0, 4), (0, 0)),
'constant')[4:].reshape(1, 1, *dp.shape)
signal = ElectronDiffraction2D(shifted)
spg = SubpixelrefinementGenerator(signal, np.array([[0, 0]]))
peaks = spg.conventional_xc(100, 20, 1).data[0, 0,
0] # as quoted in the issue
np.testing.assert_allclose([0, -4], peaks)
""" we also test com method for clarity """
peaks = spg.center_of_mass_method(60).data[0, 0, 0]
np.testing.assert_allclose([0, -4], peaks, atol=1.5)
def test_failure_for_non_even_entry_to_get_simulated_disc():
with pytest.raises(ValueError, match="'square_size' must be an even number"):
disc = get_simulated_disc(61, 5)