Merge pull request #98 from SMTG-Bham/bug-fix

fix multiphon error for different sized x axes

tests/test_plot/test_phonons.py

@@ -117,7 +117,7 @@ def test_bandmin(self):
  self.ax.set_xlabel.assert_called_once()
  self.ax.set_ylabel.assert_called_once()
  self.ax.set_xlim.assert_called_once_with(0, 1)
- self.ax.set_ylim.assert_not_called()
+ self.ax.set_ylim.assert_called_once_with(bottom=0)
 
  def test_bandmax(self):
  self.ax.spines['bottom'].get_linewidth().return_value = 1

tp/plot/phonons.py

@@ -346,11 +346,15 @@ def add_multi(ax, data, bandmin=None, bandmax=None, main=True, label=None,
  else:
  bandmax = np.amin([len(data[0]['frequency'][0]), bandmax])
 
- f = [d['frequency'] for d in data]
- f = np.array(f)[:,:,bandmin:bandmax]
-
- if round(np.amin(f), 1) == 0:
- ax.set_ylim(bottom=0)
+ if bandmin < 3:
+ f = [d['frequency'] for d in data]
+ noim = True
+ for ff in f:
+ ff = np.array(ff)[:,bandmin:bandmax]
+ if round(np.amin(ff), 1) < 0:
+ noim = False
+ if noim == True:
+ ax.set_ylim(bottom=0)
  formatting(ax, data[0], 'frequency', **xmarkkwargs)
 
  return