cleanup

pulse_lib/segments/conditional_segment.py

@@ -169,8 +169,3 @@ class conditional_segment:
     def get_metadata(self):
         pass
 
-
-if __name__ == '__main__':
-    import pulse_lib.segments.utility.looping as lp
-    import matplotlib.pyplot as plt
-

pulse_lib/segments/data_classes/data_generic.py

@@ -41,19 +41,19 @@ class parent_data(ABC):
 
     @abstractmethod
     def append():
-        raise NotImplemented
+        raise NotImplementedError()
 
     @abstractmethod
     def slice_time():
-        raise NotImplemented
+        raise NotImplementedError()
 
     @abstractmethod
     def reset_time(time = None, extend_only = False):
-        raise NotImplemented
+        raise NotImplementedError()
 
     @abstractmethod
     def wait(time):
-        raise NotImplemented
+        raise NotImplementedError()
 
 #    @abstractmethod
 #    def get_vmax(self,sample_rate):
@@ -82,26 +82,26 @@ class parent_data(ABC):
         Returns:
             integrate (double) : the integrated value of the waveform (unit is mV/sec).
         '''
-        raise NotImplemented
+        raise NotImplementedError()
 
     @abstractmethod
     def __add__():
-        raise NotImplemented
+        raise NotImplementedError()
 
     @abstractmethod
     def __mul__():
-        raise NotImplemented
+        raise NotImplementedError()
 
     @abstractmethod
     def __copy__():
-        raise NotImplemented
+        raise NotImplementedError()
 
     @abstractmethod
     def _render(self, sample_rate, ref_channel_states):
         '''
         make a full rendering of the waveform at a predetermined sample rate. This should be defined in the child of this class.
         '''
-        raise NotImplemented
+        raise NotImplementedError()
 
     def add_software_marker(self, marker_name, time):
         '''

pulse_lib/segments/data_classes/data_pulse.py

@@ -836,37 +836,3 @@ class pulse_data(parent_data):
 
         return metadata
 
-if __name__ == '__main__':
-    """
-    test functions for the IQ_data object
-    """
-    from pulse_lib.segments.data_classes.data_IQ import IQ_data_single
-    import matplotlib.pyplot as plt
-    # make two shaped pulses after each other.
-    IQ_data_object1 = IQ_data_single()
-    IQ_data_object1.start = 10
-    IQ_data_object1.stop = 140
-    IQ_data_object1.amplitude = 1
-    IQ_data_object1.frequency = 1e8
-    IQ_data_object1.start_phase = np.pi/2
-    IQ_data_object1.envelope = envelope_generator('flattop')
-
-    IQ_data_object2 = IQ_data_single()
-    IQ_data_object2.start = 150
-    IQ_data_object2.stop = 190
-    IQ_data_object2.amplitude = 1
-    IQ_data_object2.frequency = 1e8
-
-    data = pulse_data()
-    data.add_MW_data(IQ_data_object1)
-    data.reset_time(data.total_time)
-    data.add_MW_data(IQ_data_object2)
-    data.reset_time(data.total_time)
-    data.add_pulse_data(base_pulse_element(0,140, 1 , 1))
-    data.add_pulse_data(base_pulse_element(200,280, 2 , 1))
-
-    data.repeat(2)
-    rendering_data = data.render(1e9)
-    t = np.linspace(0, data.total_time, len(rendering_data))
-    plt.plot(t, rendering_data)
-    plt.show()
\ No newline at end of file