lattice/pyerrors
1
0
Fork 0
mirror of https://github.com/fjosw/pyerrors.git synced 2025-10-20 11:25:46 +02:00
Code Issues Projects Releases Packages Wiki Activity

pyerrors/correlators.py: Allowing for None values in pe.Corr.prune (#272)

Browse source 
* pyerrors/correlators.py: Allowing for None values in pe.Corr.prune

Closes: fjosw#271

* wip

* test/correlators_test.py: Adding test for updated prune method.

Closes #271
This commit is contained in:
Ruaidhrí 2025-10-19 11:28:36 +01:00 • committed by GitHub
commit e0bfcabc0c
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
2 changed files with 31 additions and 6 deletions
Download patch file Download diff file Expand all files Collapse all files

23
tests/correlators_test.py
View file

@ -781,3 +781,26 @@ def test_complex_add_and_mul():
cc += 2j
cc = cc * 4j
cc.real + cc.imag
def test_prune_with_Nones():
N = 3
T = 10
front_padding = 1
back_padding = T // 2
Ntrunc = N - 1
t0proj = 2
tproj = 3
corr_content = np.array([[[pe.pseudo_Obs((i+j+1)**(-t), .01, "None_prune_test") for i in range(N)] for j in range(N)] for t in range(T // 2 - front_padding)])
unpadded_corr = pe.Corr(corr_content)
padded_corr = pe.Corr(corr_content, padding=[front_padding, back_padding])
tmp_corr = unpadded_corr.prune(Ntrunc, t0proj=t0proj-front_padding, tproj=tproj-front_padding)
pruned_then_padded = pe.Corr(tmp_corr.content, padding=[front_padding, back_padding])
padded_then_pruned = padded_corr.prune(Ntrunc, t0proj=t0proj, tproj=tproj)
for t in range(T):
assert np.all(pruned_then_padded.content[t] == padded_then_pruned.content[t])