Repository: wright-group/WrightTools

Issue:
better WMEL defaults



Discussion:

Resolving patch:
diff --git a/WrightTools/diagrams/WMEL.py b/WrightTools/diagrams/WMEL.py
--- a/WrightTools/diagrams/WMEL.py
+++ b/WrightTools/diagrams/WMEL.py
@@ -15,9 +15,8 @@ class Subplot:
 
     def __init__(self, ax, energies, number_of_interactions=4, title='',
                  title_font_size=16, state_names=None, virtual=[None],
-                 state_font_size=8, state_text_buffer=0.5, label_side='left'):
-        '''
-        Subplot.
+                 state_font_size=14, state_text_buffer=0.5, label_side='left'):
+        """ Subplot.
 
         Parameters
         ----------
@@ -37,7 +36,7 @@ def __init__(self, ax, energies, number_of_interactions=4, title='',
             font size for the state lables
         state_text_buffer: numtype (optional)
             space between the energy level bars and the state labels
-        '''
+        """
         self.ax = ax
         self.energies = energies
         self.interactions = number_of_interactions
@@ -73,10 +72,9 @@ def __init__(self, ax, energies, number_of_interactions=4, title='',
         # title
         self.ax.set_title(title, fontsize=title_font_size)
 
-    def add_arrow(self, index, between, kind, label='', head_length=0.075,
-                  head_aspect=2, font_size=7, color='k'):
-        '''
-        Add an arrow to the WMEL diagram.
+    def add_arrow(self, index, between, kind, label='', head_length=0.1,
+                  head_aspect=2, font_size=14, color='k'):
+        """ Add an arrow to the WMEL diagram.
 
         Parameters
         ----------
@@ -98,7 +96,7 @@ def add_arrow(self, index, between, kind, label='', head_length=0.075,
         Returns
         -------
         [line,arrow_head,text]
-        '''
+        """
         if hasattr(index, 'index'):
             x_pos = list(index)
         else:
@@ -140,7 +138,7 @@ def add_arrow(self, index, between, kind, label='', head_length=0.075,
                                    fc=color, ec=color, linestyle='solid',
                                    linewidth=0, zorder=10)
         # add text
-        text = self.ax.text(np.mean(x_pos), -0.1, label, fontsize=font_size,
+        text = self.ax.text(np.mean(x_pos), -0.15, label, fontsize=font_size,
                             horizontalalignment='center')
         return line, arrow_head, text
 
@@ -153,9 +151,7 @@ class Artist:
     def __init__(self, size, energies, state_names=None,
                  number_of_interactions=4, virtual=[None],
                  state_font_size=8, state_text_buffer=0.5):
-        '''
-        virtual a list of indicies
-        '''
+        """ virtual a list of indicies """
         # create figure
         figsize = [int(size[0] * ((number_of_interactions + 1.) / 6.)), size[1] * 2.5]
         fig, (subplots) = plt.subplots(size[1], size[0], figsize=figsize)
@@ -207,10 +203,10 @@ def clear_diagram(self, diagram):
 
     def add_arrow(self, diagram, number, between, kind, label='',
                   head_length=0.075, font_size=7, color='k'):
-        '''
+        """
         kind one in [ket, bra, out] \n
         returns [line, arrow_head, text]
-        '''
+        """
         column, row = diagram
         x_pos = self.x_pos[number]
         # calculate arrow length


Test patch: