Fix PEP8 issues

tardis-sn · Jun 6, 2017 · ae46cc3 · ae46cc3
1 parent 6cd9bdb
commit ae46cc3
Show file tree

Hide file tree

Showing 2 changed files with 11 additions and 4 deletions.
diff --git a/tardis/montecarlo/spectrum.py b/tardis/montecarlo/spectrum.py
@@ -93,18 +93,25 @@ def f_nu_to_f_lambda(self, f_nu):
     def plot(self, ax, mode='wavelength'):
         if mode == 'wavelength':
             ax.plot(self.wavelength.value, self.flux_lambda.value)
-            ax.set_xlabel('Wavelength [%s]' % self.wavelength.unit._repr_latex_())
+            ax.set_xlabel('Wavelength [{}]'.format(
+                self.wavelength.unit._repr_latex_())
+                )
             ax.set_ylabel('Flux [%s]' % self.flux_lambda.unit._repr_latex_())
 
     def to_ascii(self, fname, mode='luminosity_density'):
         if mode == 'luminosity_density':
-            np.savetxt(fname, zip(self.wavelength.value, self.luminosity_density_lambda.value))
+            np.savetxt(
+                    fname, zip(
+                        self.wavelength.value,
+                        self.luminosity_density_lambda.value))
         elif mode == 'flux':
             np.savetxt(
                     fname,
                     zip(self.wavelength.value, self.flux_lambda.value))
         else:
-            raise NotImplementedError('only mode "luminosity_density" and "flux" are implemented')
+            raise NotImplementedError(
+                    'only mode "luminosity_density"'
+                    'and "flux" are implemented')
 
     def to_hdf(self, path_or_buf, path):
         pass

diff --git a/tardis/montecarlo/tests/test_spectrum.py b/tardis/montecarlo/tests/test_spectrum.py
@@ -1,4 +1,3 @@
-import os
 import pytest
 import numpy as np
 import pandas as pd
@@ -189,6 +188,7 @@ def test_to_from_hdf_buffer(tmpdir, spectrum):
             spectrum,
             spec_from)
 
+
 def test_creat_from_wl(spectrum):
     actual = TARDISSpectrum(
             spectrum._frequency.to('angstrom', u.spectral()),