fix: Corr.__repr__ also works for complex correlators.

This commit is contained in:
Fabian Joswig 2022-12-20 17:40:41 +01:00
parent f68a5d044f
commit f39a70c5a8
No known key found for this signature in database
2 changed files with 3 additions and 0 deletions

View file

@ -969,6 +969,8 @@ class Corr:
content_string += "Description: " + self.tag + "\n"
if self.N != 1:
return content_string
if isinstance(self[0], CObs):
return content_string
if print_range[1]:
print_range[1] += 1

View file

@ -539,3 +539,4 @@ def test_complex_Corr():
cobs = pe.CObs(o1, o1)
ccorr = pe.Corr([cobs, cobs, cobs])
assert np.all([ccorr.imag[i] == ccorr.real[i] for i in range(ccorr.T)])
print(ccorr)