Skip to content

Commit d976243

Browse files
committed
Check flm zero pattern in test
1 parent 6263304 commit d976243

File tree

1 file changed

+8
-0
lines changed

1 file changed

+8
-0
lines changed

tests/test_signal_generator.py

Lines changed: 8 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -42,6 +42,13 @@ def test_complex_el_and_m_indices(L, min_el):
4242
assert (m_indices == expected_m_indices).all()
4343

4444

45+
def check_flm_zeros(flm, L, min_el):
46+
for el in range(L):
47+
for m in range(L):
48+
if el < min_el or m > el:
49+
assert flm[el, L - 1 + m] == flm[el, L - 1 - m] == 0
50+
51+
4552
def check_flm_conjugate_symmetry(flm, L, min_el):
4653
for el in range(min_el, L):
4754
for m in range(el + 1):
@@ -60,6 +67,7 @@ def test_generate_flm(rng, L, L_lower, spin, reality):
6067
assert flm.shape == smp.s2_samples.flm_shape(L)
6168
assert flm.dtype == np.complex128
6269
assert np.isfinite(flm).all()
70+
check_flm_zeros(flm, L, max(L_lower, abs(spin)))
6371
if reality:
6472
check_flm_conjugate_symmetry(flm, L, max(L_lower, abs(spin)))
6573
f_complex = s2fft.inverse(flm, L, spin=spin, reality=False, L_lower=L_lower)

0 commit comments

Comments
 (0)