diff --git a/tangostationcontrol/test/devices/sdp/test_beamlet_device.py b/tangostationcontrol/test/devices/sdp/test_beamlet_device.py index 0e8ef1a105c09ef2540c1268348b994696323f9f..f5166079e730e41c39390b06ced286af9716f95e 100644 --- a/tangostationcontrol/test/devices/sdp/test_beamlet_device.py +++ b/tangostationcontrol/test/devices/sdp/test_beamlet_device.py @@ -177,7 +177,12 @@ class TestBeamletDevice(base.TestCase): def test_beamlet_frequencies_basic(self): beamlet_subbands = numpy.array( - [[[list(range(N_beamlets_ctrl))] * N_pol] * A_pn] * N_pn, dtype=numpy.uint32 + [ + [[beamlet % N_subbands for beamlet in range(N_beamlets_ctrl)] * N_pol] + * A_pn + ] + * N_pn, + dtype=numpy.uint32, ) subband_indices = numpy.array( [[[list(range(N_subbands))] * N_pol] * A_pn] * N_pn, dtype=numpy.uint32 @@ -190,7 +195,12 @@ class TestBeamletDevice(base.TestCase): """Test different settings for the FPGAs to verify the ordering.""" beamlet_subbands = numpy.array( - [[[list(range(N_beamlets_ctrl))] * N_pol] * A_pn] * N_pn, dtype=numpy.uint32 + [ + [[beamlet % N_subbands for beamlet in range(N_beamlets_ctrl)] * N_pol] + * A_pn + ] + * N_pn, + dtype=numpy.uint32, ) subband_indices = numpy.array( [[[list(range(N_subbands))] * N_pol] * A_pn] * N_pn, dtype=numpy.uint32 @@ -206,7 +216,12 @@ class TestBeamletDevice(base.TestCase): """Test different settings for the inputs of the FPGAs to verify the ordering.""" beamlet_subbands = numpy.array( - [[[list(range(N_beamlets_ctrl))] * N_pol] * A_pn] * N_pn, dtype=numpy.uint32 + [ + [[beamlet % N_subbands for beamlet in range(N_beamlets_ctrl)] * N_pol] + * A_pn + ] + * N_pn, + dtype=numpy.uint32, ) subband_indices = numpy.array( [[[list(range(N_subbands))] * N_pol] * A_pn] * N_pn, dtype=numpy.uint32 @@ -222,7 +237,12 @@ class TestBeamletDevice(base.TestCase): """Test a shuffled order of subbands.""" beamlet_subbands = numpy.array( - [[[list(range(N_beamlets_ctrl))] * N_pol] * A_pn] * N_pn, dtype=numpy.uint32 + [ + [[beamlet % N_subbands for beamlet in range(N_beamlets_ctrl)] * N_pol] + * A_pn + ] + * N_pn, + dtype=numpy.uint32, ) subband_indices = numpy.array( [[[list(range(N_subbands))] * N_pol] * A_pn] * N_pn, dtype=numpy.uint32