Propose fix for nc/idl map (#284)

* proposed fix for nc/idl map

* correct docs, implement failesafe for therm. trajectories

* lint

* easier test, improve descriptions

* clarify r_step parameterin docs

* add small test for trajectories
This commit is contained in:
Justus Kuhlmann 2026-05-28 11:37:21 +02:00 committed by GitHub
commit 30d109413d
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
3 changed files with 40 additions and 5 deletions

View file

@ -65,6 +65,37 @@ def test_rwms():
pe.input.openQCD.extract_t0(path, '', dtr_read=3, xmin=0, spatial_extent=4, files=files, names=names, fit_range=2, plot_fit=True)
# trajectories
t0 = pe.input.openQCD.extract_t0(path, 'oqcd2_traj', dtr_read=30, xmin=16, spatial_extent=48, fit_range=2, plot_fit=True, names = ["A|r1"], assume_thermalization=False)
assert len(t0.idl['A|r1']) == 10
assert t0.idl['A|r1'][0] == 4
assert t0.idl['A|r1'][9] == 13
with pytest.warns(Warning):
t0 = pe.input.openQCD.extract_t0(path, 'oqcd2_traj', dtr_read=1, xmin=16, spatial_extent=48, fit_range=2, plot_fit=True, names = ["A|r1"])
assert len(t0.idl['A|r1']) == 30
assert t0.idl['A|r1'][0] == 1
assert t0.idl['A|r1'][29] == 30
with pytest.warns(Warning):
t0 = pe.input.openQCD.extract_t0(path, 'oqcd2_traj', dtr_read=10, xmin=16, spatial_extent=48, fit_range=2, plot_fit=True, names = ["A|r1"])
assert len(t0.idl['A|r1']) == 30
assert t0.idl['A|r1'][0] == 1
assert t0.idl['A|r1'][29] == 30
with pytest.warns(Warning):
t0 = pe.input.openQCD.extract_t0(path, 'oqcd2_traj', dtr_read=30, xmin=16, spatial_extent=48, fit_range=2, plot_fit=True, names = ["A|r1"])
assert len(t0.idl['A|r1']) == 10
assert t0.idl['A|r1'][0] == 1
assert t0.idl['A|r1'][9] == 10
with pytest.warns(Warning):
t0 = pe.input.openQCD.extract_t0(path, 'oqcd2_traj', dtr_read=60, xmin=16, spatial_extent=48, fit_range=2, plot_fit=True, names = ["A|r1"])
assert len(t0.idl['A|r1']) == 5
assert t0.idl['A|r1'][0] == 1
assert t0.idl['A|r1'][4] == 5
with pytest.raises(Exception):
pe.input.openQCD.extract_t0(path, '', dtr_read=3, xmin=0, spatial_extent=4, files=files, names=names, fit_range=2, c=14)
# w0