fix a bug in find_pulse

qutip · May 2, 2021 · a57e3c2 · a57e3c2
1 parent c11dc37
commit a57e3c2
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 2 deletions.
diff --git a/src/qutip_qip/device/processor.py b/src/qutip_qip/device/processor.py
@@ -244,9 +244,10 @@ def get_pulse_dict(self):
         return label_list
 
     def find_pulse(self, pulse_name):
+        pulse_dict = self.get_pulse_dict()
         if isinstance(pulse_name, str):
             try:
-                return self.pulses[self.pulse_dict[pulse_name]]
+                return self.pulses[pulse_dict[pulse_name]]
             except (KeyError):
                 raise KeyError(
                     "Pulse name {} undefined. "

diff --git a/tests/test_processor.py b/tests/test_processor.py
@@ -342,7 +342,7 @@ def testChooseSolver(self):
         result = proc.run_state(init_state=init_state, solver="mcsolve")
         assert_allclose(
             fidelity(result.states[-1], qubit_states(2, [0, 1, 0, 0])),
-            1, rtol=1.e-7) 
+            1, rtol=1.e-7)
 
     def test_no_saving_intermidiate_state(self):
         processor = Processor(1)
@@ -352,3 +352,8 @@ def test_no_saving_intermidiate_state(self):
             )
         result = processor.run_state(basis(2,0), tlist=[0,1])
         assert(len(result.states) == 2)
+
+    def test_pulse_dict(self):
+        processor = Processor(1)
+        processor.add_control(sigmax(), 0, label="test")
+        assert("test" in processor.get_pulse_dict())