Update doctest values to match updated Cell calculations.

Author: dihm

src/rydiqule/cell.py

@@ -746,7 +746,8 @@ def add_single_coupling(
         Coherent Couplings: 
             ((5, 0, 0.5),(5, 1, 1.5)): {rabi_frequency: 2.0, detuning: 1.0, phase: 0, kvec: (0, 0, 0), label: probe, coherent_cc: 1, dipole_moment: 2.44, q: 0}
         Decoherent Couplings:
-            ((5, 1, 1.5),(5, 0, 0.5)): {gamma_transition: 38.11316}
+            ((5, 0, 0.5),(5, 0, 0.5)): {gamma_transit: 0.40697}
+            ((5, 1, 1.5),(5, 0, 0.5)): {gamma_transition: 38.11316, gamma_transit: 0.40697}
         Energy Shifts:
             None
         
@@ -764,7 +765,8 @@ def add_single_coupling(
         Coherent Couplings: 
             ((5, 0, 0.5),(5, 1, 1.5)): {rabi_frequency: 2.0, detuning: 1.0, phase: 0, kvec: (0, 0, 0), label: probe, coherent_cc: 1, dipole_moment: 2.44, q: 0}
         Decoherent Couplings:
-            ((5, 1, 1.5),(5, 0, 0.5)): {gamma_transition: 38.11}
+            ((5, 0, 0.5),(5, 0, 0.5)): {gamma_transit: 0.40696}
+            ((5, 1, 1.5),(5, 0, 0.5)): {gamma_transition: 38.113, gamma_transit: 0.40696}
         Energy Shifts:
             None
 
@@ -781,7 +783,8 @@ def add_single_coupling(
         Coherent Couplings: 
             ((5, 0, 0.5),(5, 1, 1.5)): {rabi_frequency: 1.177, detuning: 1.0, phase: 0, kvec: (0, 0, 0), label: probe, coherent_cc: 1, dipole_moment: 2.44, q: 0}
         Decoherent Couplings:
-            ((5, 1, 1.5),(5, 0, 0.5)): {gamma_transition: 38.11}
+            ((5, 0, 0.5),(5, 0, 0.5)): {gamma_transit: 0.40696}
+            ((5, 1, 1.5),(5, 0, 0.5)): {gamma_transition: 38.11, gamma_transit: 0.40696}
         Energy Shifts:
             None
 
@@ -796,7 +799,8 @@ def add_single_coupling(
         Coherent Couplings: 
             ((5, 0, 0.5),(5, 1, 1.5)): {rabi_frequency: 4.3, detuning: 1.0, phase: 0, kvec: (0, 0, 0), label: probe, coherent_cc: 1, dipole_moment: 2.44, q: 0}
         Decoherent Couplings:
-            ((5, 1, 1.5),(5, 0, 0.5)): {gamma_transition: 38.11}
+            ((5, 0, 0.5),(5, 0, 0.5)): {gamma_transit: 0.4117}
+            ((5, 1, 1.5),(5, 0, 0.5)): {gamma_transition: 38.113, gamma_transit: 0.4117}
         Energy Shifts:
             None
 

src/rydiqule/experiments.py

@@ -90,8 +90,8 @@ def get_snr(sensor: Sensor,
     >>> c.add_coupling(states=(g,e), rabi_frequency=np.linspace(1e-6, 1, 5), detuning=1, label="probe")
     >>> snr, mesh = rq.get_snr(c, 'probe_rabi_frequency')
     >>> print(snr) 
-    [       0.       13947396.7 27887614.4 41813486.6
-     55717871.1]
+    [       0.       13654034.1 27301261.5 40934886.9
+     54548137.6]
     >>> print(mesh) # doctest: +SKIP
     [array([0.      , 0.25    , 0.499999, 0.749999, 0.999999])]
 

src/rydiqule/sensor_solution.py

@@ -352,7 +352,7 @@ def get_solution_element(self, idx: int) -> Result:
         (3,)
         >>> rho_01_im = sols.get_solution_element(0)
         >>> print(rho_01_im)
-        0.0013711656
+        0.00131399
 
         """
         basis_size = np.sqrt(self.rho.shape[-1] + 1)
@@ -382,7 +382,11 @@ def coupling_coefficient_matrix(self, coupling: sensor_utils.StateSpecs) -> np.n
         >>> sol = rq.solve_steady_state(my_cell)
         >>> for e in sol.couplings.edges(data="coherent_cc"):
         ...     print(*e)
+        (5, 0, 0.5, m_j=-0.5) (5, 0, 0.5, m_j=-0.5) None
+        (5, 0, 0.5, m_j=-0.5) (5, 0, 0.5, m_j=0.5) None
         (5, 0, 0.5, m_j=-0.5) (5, 1, 0.5, m_j=-0.5) -0.816496580927726
+        (5, 0, 0.5, m_j=0.5) (5, 0, 0.5, m_j=-0.5) None
+        (5, 0, 0.5, m_j=0.5) (5, 0, 0.5, m_j=0.5) None
         (5, 0, 0.5, m_j=0.5) (5, 1, 0.5, m_j=0.5) 0.816496580927726
         (5, 1, 0.5, m_j=-0.5) (5, 0, 0.5, m_j=-0.5) None
         (5, 1, 0.5, m_j=-0.5) (5, 0, 0.5, m_j=0.5) None
@@ -495,7 +499,7 @@ def get_susceptibility(self) -> ComplexResult:
         (3,)
         >>> sus = sols.get_susceptibility()
         >>> print(sus)
-        (1.9734254e-05+0.000376067j)
+        (1.891136e-05+0.00036817j)
 
         """
         probe_rabi = self.probe_rabi*1e6 #rad/s
@@ -539,7 +543,7 @@ def get_OD(self) -> Result:
         (3,)
         >>> OD = sols.get_OD()
         >>> print(OD)
-        0.3028422896
+        0.2964844
 
         """
 
@@ -577,7 +581,7 @@ def get_transmission_coef(self) -> Result:
         (3,)
         >>> t = sols.get_transmission_coef()
         >>> print(t)
-        0.73871559029
+        0.743427
 
         """
         OD = self.get_OD()
@@ -605,7 +609,7 @@ def get_phase_shift(self) -> Result:
         >>> print(sols.rho.shape)
         (3,)
         >>> print(sols.get_phase_shift())
-        80.5295218956
+        80.5294887
         
         """
         # reverse probe tuple order to get correct sign of imag

src/rydiqule/sensor_utils.py

@@ -405,9 +405,9 @@ def convert_to_full_dm(dm: np.ndarray) -> np.ndarray:
     >>> c.add_coupling(states=(g, e), rabi_frequency=1, detuning=1)
     >>> sols = rq.solve_steady_state(c)
     >>> print(sols.rho)
-    [0.001371 0.02613  0.000686]
+    [0.001313 0.02558  0.000664]
     >>> print(rq.sensor_utils.convert_to_full_dm(sols.rho))
-    [9.993144e-01 1.371165e-03 2.612972e-02 6.855828e-04]
+    [9.993359e-01 1.31399e-03 2.55811e-02 6.64016e-04]
 
     """
     b, r = divmod(math.sqrt(dm.shape[-1]+1), 1.0)
@@ -461,10 +461,10 @@ def convert_dm_to_complex(dm: np.ndarray) -> np.ndarray:
     >>> c.add_coupling(states=(g, e), rabi_frequency=1, detuning=1)
     >>> sols = rq.solve_steady_state(c)
     >>> print(sols.rho)
-    [0.001371 0.02613  0.000686]
+    [0.001313 0.02558  0.000664]
     >>> print(rq.sensor_utils.convert_dm_to_complex(sols.rho))
-    [[9.993144e-01+0.j      1.371166e-03+0.02613j]
-     [1.371166e-03-0.02613j 6.855828e-04+0.j     ]]
+    [[9.993359e-01+0.j      1.31399e-03+0.025581j]
+     [1.313990e-03-0.025581j 6.64016e-04+0.j     ]]
 
     """
     b, r = divmod(math.sqrt(dm.shape[-1]+1), 1.0)