Refractive index corrected

calculation.py

@@ -67,7 +67,7 @@ def mqw_psi(lz=0.01e-9, electric_field=0., wavelength=850e-9):
 
 def benchmark_refractive_index():
     al = np.array([0.0, 0.2, 0.4, 0.6, 0.8, 1.0])
-    wavelength = np.arange(500e-9, 1500e-9, 10e-9)
+    wavelength = np.arange(500e-9, 1500e-9, 1e-9)
     indices_array = np.empty([al.size, wavelength.size])
 
     for x in range(al.size):

optic.py

@@ -242,17 +242,17 @@ def afromovitz_varshni_real_algaas_refractive_index(al, wavelength):
     e_g = (1 -al)*1.424 +al*3.02 -al*(1-al)*0.37
 
     # Varshni
-    alpha = 5.405 +4.038*al*1e-4                                        # [eV/K]
-    beta = 204/370*(370 +54*al +22*al**2)                               # [K]
-    e_var = e_g +alpha*298**2/(298 +beta) -(alpha*T**2)/(T +beta)       # [eV?]
+    alpha = (5.405 +4.038*al)*1e-4                                      # [eV/K]
+    beta = 204./370.*(370. +54.*al +22.*al**2)                          # [K]
+    e_var = e_g +alpha*298.**2/(298. +beta) -(alpha*T**2)/(T +beta)     # [eV?]
 
     # Afromovitz
     energy = e +1j*1e-2
 
-    n = 1 +e_d/e_1
+    n = 1. +e_d/e_1
     n += (e_d/(e_1**3))*energy**2
-    n_l = (e_d/(2*e_1**3*(e_1**2 -e_var**2)))*energy**4
-    n_l *= np.log((2*e_1**2 -e_var**2 -energy**2)/(e_var**2 -energy**2))
+    n_l = (e_d/(2.*e_1**3*(e_1**2 -e_var**2)))*energy**4
+    n_l *= np.log((2.*e_1**2 -e_var**2 -energy**2)/(e_var**2 -energy**2))
 
     n += n_l
     n = np.real(np.sqrt(n))