Bugfix

Some obvious ones

Author: lcrippa

src/edipack2py/func_aux_funx.py

@@ -367,14 +367,14 @@ def get_ed_mode(self):
 
 
 # set phonons displacement and coupling
-def set_phonon_coefficients(self,disp=None,coupling=None):
+def set_phonon_coefficients(self, displacement=None, coupling=None):
     """
 
     This function sets new values for the phonon displacement field
     and the electron-phonon coupling matrix
     
-    :type disp: float 
-    :param disp: phonon displacement field
+    :type displacement: float 
+    :param displacement: phonon displacement field
     
     :type coupling: np.array(float) 
     :param coupling: electron-phonon coupling matrix of shape  ( :code:`Norb` ) or ( :code:`Norb`, :code:`Norb` ) 
@@ -385,21 +385,23 @@ def set_phonon_coefficients(self,disp=None,coupling=None):
     set_a_ph.argtypes = [np.ctypeslib.ndpointer(dtype=float, ndim=1, flags="F_CONTIGUOUS")]
     set_a_ph.restype = None
 
-    set_g_ph = self.library.ed_setG_ph
+    set_g_ph = self.library.ed_set_G_ph
     set_g_ph.argtypes = [np.ctypeslib.ndpointer(dtype=float, ndim=2, flags="F_CONTIGUOUS")]
     set_g_ph.restype = None
 
     aux_norb = ct.c_int.in_dll(self.library, "Norb").value
 
-    if disp==None and coupling==None:
+    if displacement==None and coupling==None:
         raise RuntimeError("set_phonon_coefficients: provide at least either disp or coupling")
 
-    if disp != None:
-        set_a_ph(np.array([disp]))
+    if displacement != None:
+        set_a_ph(np.array([displacement],order="F"))
+        
     if coupling != None:
         if np.shape(copuling) == (aux_norb,):
             coupling = np.array(np.diag(coupling),order="F")
         elif np.shape(copuling) != (aux_norb,aux_norb):
             raise ValueError("set_phonon_coefficients: coupling shape is (Norb) or (Norb,Norb)")
         set_g_ph(coupling)
-        
+    
+    return

src/edipack2py/func_io.py

@@ -1491,7 +1491,7 @@ def get_dimp(self, zeta=None, axis=None):
             elif axis == "r":
                 axisflag = 1
             else:
-                raise ValueError("axis can only be m, r or t")
+                raise ValueError("axis can only be 'm' or 'r'")
             nfreq = np.shape(zeta)[0]
 
     zeta = np.asfortranarray(zeta)
@@ -1539,10 +1539,7 @@ def get_denmat(self, ishape=4, doprint=False):
     ]
     ed_get_denmat_n4.restype = None
 
-    if doprint:
-        doprintflag=1
-    else:
-        doprintflag=0
+    doprint = int(doprint)
 
     if self.Nineq != 0:
         raise RuntimeError("get_denmat not implemented for inequivalent sites yet.")
@@ -1553,24 +1550,23 @@ def get_denmat(self, ishape=4, doprint=False):
 
     bath_type = self.get_bath_type()
 
-    if bath_type == 1 or bath_type == 3:
+    if bath_type == 1 or bath_type > 3:
         Ns = (aux_nbath + 1) * aux_norb
     elif bath_type == 2:
         Ns = aux_nbath + aux_norb
     else:
-        raise ValueError("get_denmat: wrong bath type")
-        
+        raise ValueError("get_denmat: wrong bath type")      
 
     if ishape == 4:
         denmat = np.zeros([aux_nspin,aux_nspin,Ns,Ns], dtype=complex, order="F")
-        ed_get_denmat_n4(denmat,doprintflag)
+        ed_get_denmat_n4(denmat,doprint)
     elif ishape == 2:
         denmat = np.zeros([aux_nspin*Ns,aux_nspin*Ns], dtype=complex, order="F")
-        ed_get_denmat_n2(denmat,doprintflag)
+        ed_get_denmat_n2(denmat,doprint)
     else:
         raise ValueError("get_denmat: wrong ishape")
 
 
     denmat = np.ascontiguousarray(denmat)
 
-    return dimp
+    return denmat