1 files changed, 8 insertions, 3 deletions
diff --git a/test.m b/test.m
index 5c8de4c..17a1622 100644
--- a/test.m
+++ b/test.m
@@ -158,18 +158,23 @@ L=Liouville_operator_matrix(
 %solving for density matrix vector
 rhoLiouville=L\rhoLiouville_dot;
 
+rho_2=rhoOfFreq(rhoLiouville, 2, Nlevels);
+rho_1=rhoOfFreq(rhoLiouville, 1, Nlevels);
 
 L_new=L;
 rhoLiouville_new=rhoLiouville;
+rho_2_new=rho_2;
+rho_1_new=rho_1;
 
 % uncomment to update reference file
-%save 'L_and_rhoL_referenced.mat' L rhoLiouville
+% save 'L_and_rhoL_referenced.mat' L rhoLiouville rho_1 rho_2
 
-clear L rhoLiouville;
+clear L rhoLiouville rho_2 rho_1;
 
 load 'L_and_rhoL_referenced.mat' 
 diff_with_reference_L=sum(sum(abs(L_new-L)))
-
 diff_with_reference_rhoL=(sum(abs(rhoLiouville_new -rhoLiouville)))
+diff_with_rho_1=(sum(sum(abs(rho_1_new -rho_1))))
+diff_with_rho_2=(sum(sum(abs(rho_2_new -rho_2))))
 
 elapsed_time = etime (clock (), t0)