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

Closes #271
This commit is contained in:
campioru 2025-10-18 19:07:43 +01:00
commit ff03d8a06c

View file

@ -800,8 +800,7 @@ def test_prune_with_Nones():
tmp_corr = unpadded_corr.prune(Ntrunc, t0proj=t0proj-front_padding, tproj=tproj-front_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]) pruned_then_padded = pe.Corr(tmp_corr.content, padding=[front_padding, back_padding])
with pytest.raises(ValueError): padded_then_pruned = padded_corr.prune(Ntrunc, t0proj=t0proj, tproj=tproj)
padded_then_pruned = padded_corr.prune(Ntrunc, t0proj=t0proj, tproj=tproj)
for t in range(T): for t in range(T):
assert np.all(pruned_then_padded.content[t] == padded_then_pruned.content[t]) assert np.all(pruned_then_padded.content[t] == padded_then_pruned.content[t])