Fix test for modulation period for both heterodyne and baseband cases

Author: gitj

kid_readout/roach/interface.py

@@ -230,6 +230,10 @@ def state(self):
         return self.get_state()
 
     def get_state(self,include_registers=False):
+        if self.phase0 is None:
+            reference_sequence_number = -1
+        else:
+            reference_sequence_number = int(self.phase0)
         roach_state = StateDict(boffile=self.boffile,
                           heterodyne=self.heterodyne,
                           adc_sample_rate=self.fs*1e6, # roach still uses MHz, so convert to Hz
@@ -246,7 +250,7 @@ def get_state(self,include_registers=False):
                           debug_register=self.debug_register,
                           fft_shift_register = self.fft_shift_register,
                           hardware_delay_samples = self.hardware_delay_samples,
-                          reference_sequence_number = int(self.phase0)
+                          reference_sequence_number = reference_sequence_number
                           )
         if include_registers:
             for register in self.initial_values_for_writeable_registers:

kid_readout/roach/tests/mixin.py

@@ -40,8 +40,11 @@ def test_precomputed_wavenorm(self):
             assert self.ri.wavenorm >= 0.99 * actual_wavenorm  # guarantees the wave won't overflow
 
     def test_calculate_modulation_period(self):
-        roach_state = core.StateDict(modulation_rate=7, modulation_output=2)
+        roach_state = core.StateDict(modulation_rate=7, modulation_output=2, heterodyne=True)
         assert calculate.modulation_period_samples(roach_state) == 256
+        roach_state = core.StateDict(modulation_rate=7, modulation_output=2, heterodyne=False)
+        assert calculate.modulation_period_samples(roach_state) == 128
+
 
     def test_get_measurement_blocks(self):
         num_tones = 32