diff --git a/MDANSE/Src/MDANSE/Framework/Configurators/CorrelationFramesConfigurator.py b/MDANSE/Src/MDANSE/Framework/Configurators/CorrelationFramesConfigurator.py index bc341933d..9cb7e337c 100644 --- a/MDANSE/Src/MDANSE/Framework/Configurators/CorrelationFramesConfigurator.py +++ b/MDANSE/Src/MDANSE/Framework/Configurators/CorrelationFramesConfigurator.py @@ -38,6 +38,7 @@ def configure(self, value: tuple[int, int, int, int]): first, last, step, c_frames = value super().configure((first, last, step)) + self._original_input = value if c_frames > self["n_frames"]: self.error_status = ( diff --git a/MDANSE/Tests/UnitTests/Analysis/test_scattering.py b/MDANSE/Tests/UnitTests/Analysis/test_scattering.py index dad03172d..3fc7b92e4 100644 --- a/MDANSE/Tests/UnitTests/Analysis/test_scattering.py +++ b/MDANSE/Tests/UnitTests/Analysis/test_scattering.py @@ -114,10 +114,18 @@ def test_dcsf(traj_info, qvector_grid): if any(key.startswith(j) for j in ["f(q,t)", "s(q,f)"]) ] for key in keys: - np.testing.assert_array_almost_equal( - actual[f"/{key}"] * actual[f"/{key}"].attrs["scaling_factor"], - desired[f"/{key}"], - ) + try: + scaling_factor = desired[f"/{key}"].attrs["scaling_factor"] + except KeyError: + np.testing.assert_array_almost_equal( + actual[f"/{key}"] * actual[f"/{key}"].attrs["scaling_factor"], + desired[f"/{key}"], + ) + else: + np.testing.assert_array_almost_equal( + actual[f"/{key}"] * actual[f"/{key}"].attrs["scaling_factor"], + desired[f"/{key}"] * scaling_factor, + ) os.remove(temp_name + ".mda") assert path.exists(temp_name + "_text.tar") @@ -215,10 +223,18 @@ def test_disf(traj_info, qvector_grid): i for i in desired.keys() if any([j in i for j in ["f(q,t)", "s(q,f)"]]) ] for key in keys: - np.testing.assert_array_almost_equal( - actual[f"/{key}"] * actual[f"/{key}"].attrs["scaling_factor"], - desired[f"/{key}"], - ) + try: + scaling_factor = desired[f"/{key}"].attrs["scaling_factor"] + except KeyError: + np.testing.assert_array_almost_equal( + actual[f"/{key}"] * actual[f"/{key}"].attrs["scaling_factor"], + desired[f"/{key}"], + ) + else: + np.testing.assert_array_almost_equal( + actual[f"/{key}"] * actual[f"/{key}"].attrs["scaling_factor"], + desired[f"/{key}"] * scaling_factor, + ) os.remove(temp_name + ".mda") assert path.exists(temp_name + "_text.tar") diff --git a/MDANSE/Tests/UnitTests/Results/dcsf_short_traj.mda b/MDANSE/Tests/UnitTests/Results/dcsf_short_traj.mda index 3a739ce36..60a720142 100644 Binary files a/MDANSE/Tests/UnitTests/Results/dcsf_short_traj.mda and b/MDANSE/Tests/UnitTests/Results/dcsf_short_traj.mda differ