Use f-strings where relevant

Author: amorison

stagpy/__init__.py

@@ -86,8 +86,7 @@ def load_mplstyle():
             try:
                 plt.style.use(style)
             except OSError:
-                print('Cannot import style {}.'.format(style),
-                      file=sys.stderr)
+                print(f'Cannot import style {style}.', file=sys.stderr)
                 conf.plot.mplstyle = ''
     if conf.plot.xkcd:
         plt.xkcd()

stagpy/_step.py

@@ -229,8 +229,8 @@ def __getitem__(self, name):
         else:
             raise error.UnknownFieldVarError(name)
         if parsed_data is None:
-            raise error.MissingDataError('Missing field {} in step {}'
-                                         .format(name, self.step.istep))
+            raise error.MissingDataError(
+                f'Missing field {name} in step {self.step.istep}')
         header, fields = parsed_data
         self._header = header
         for fld_name, fld in zip(fld_names, fields):
@@ -405,8 +405,8 @@ def _rprofs(self):
             step = self.step
             self._data = step.sdat._rprof_and_times[0].get(step.istep)
             if self._data is None:
-                raise error.MissingDataError('No rprof data in step {} of {}'
-                                             .format(step.istep, step.sdat))
+                raise error.MissingDataError(
+                    f'No rprof data in step {step.istep} of {step.sdat}')
         return self._data
 
     def __getitem__(self, name):

stagpy/commands.py

@@ -25,7 +25,7 @@ def info_cmd():
     sdat = stagyydata.StagyyData()
     lsnap = sdat.snaps[-1]
     lstep = sdat.steps[-1]
-    print('StagYY run in {}'.format(sdat.path))
+    print(f'StagYY run in {sdat.path}')
     if lsnap.geom.threed:
         dimension = '{} x {} x {}'.format(lsnap.geom.nxtot,
                                           lsnap.geom.nytot,
@@ -45,9 +45,9 @@ def info_cmd():
     print()
     for step in sdat.walk:
         is_snap = step.isnap is not None
-        print('Step {}/{}'.format(step.istep, lstep.istep), end='')
+        print(f'Step {step.istep}/{lstep.istep}', end='')
         if is_snap:
-            print(', snapshot {}/{}'.format(step.isnap, lsnap.isnap))
+            print(f', snapshot {step.isnap}/{lsnap.isnap}')
         else:
             print()
         series = step.timeinfo.loc[varlist]
@@ -95,7 +95,7 @@ def _pretty_print(key_val, sep=': ', min_col_width=39, text_width=None):
             wrapper.subsequent_indent = ' '
         else:
             wrapper.subsequent_indent = ' ' * (len(key) + len(sep))
-        lines.extend(wrapper.wrap('{}{}{}'.format(key, sep, val)))
+        lines.extend(wrapper.wrap(f'{key}{sep}{val}'))
 
     chunks = []
     for rem_col in range(ncols, 1, -1):
@@ -107,7 +107,7 @@ def _pretty_print(key_val, sep=': ', min_col_width=39, text_width=None):
     chunks.append(lines)
     lines = zip_longest(*chunks, fillvalue='')
 
-    fmt = '|'.join(['{{:{}}}'.format(colw)] * (ncols - 1))
+    fmt = '|'.join([f'{{:{colw}}}'] * (ncols - 1))
     fmt += '|{}' if ncols > 1 else '{}'
     print(*(fmt.format(*line) for line in lines), sep='\n')
 
@@ -157,7 +157,7 @@ def version_cmd():
 
     Use :data:`stagpy.__version__` to obtain the version in a script.
     """
-    print('stagpy version: {}'.format(__version__))
+    print(f'stagpy version: {__version__}')
 
 
 def report_parsing_problems(parsing_out):
@@ -191,7 +191,7 @@ def config_pp(subs):
                 opt += ' (c)' if meta.cmd_arg else ' (f)'
             hlp_lst.append((opt, meta.help))
         if hlp_lst:
-            print('{}:'.format(sub))
+            print(f'{sub}:')
             _pretty_print(hlp_lst, sep=' -- ',
                           text_width=min(get_terminal_size().columns, 100))
             print()

stagpy/config.py

@@ -16,7 +16,7 @@ def _actual_index(arg):
     if ':' in arg:
         idxs = arg.split(':')
         if len(idxs) > 3:
-            raise ValueError('{} is an invalid slice'.format(arg))
+            raise ValueError(f'{arg} is an invalid slice')
         idxs[0] = int(idxs[0]) if idxs[0] else None
         idxs[1] = int(idxs[1]) if idxs[1] else None
         if len(idxs) == 3:

stagpy/error.py

@@ -27,7 +27,7 @@ class NoSnapshotError(StagpyError):
 
     def __init__(self, sdat):
         self.sdat = sdat
-        super().__init__('no snapshot found for {}'.format(sdat))
+        super().__init__(f'no snapshot found for {sdat}')
 
 
 class NoParFileError(StagpyError):
@@ -43,7 +43,7 @@ class NoParFileError(StagpyError):
 
     def __init__(self, parfile):
         self.parfile = parfile
-        super().__init__('{} file not found'.format(parfile))
+        super().__init__(f'{parfile} file not found')
 
 
 class NotAvailableError(StagpyError):
@@ -130,8 +130,7 @@ class InvalidTimeFractionError(StagpyError):
 
     def __init__(self, fraction):
         self.fraction = fraction
-        super().__init__('Fraction should be in (0,1] (received {})'
-                         .format(fraction))
+        super().__init__(f'Fraction should be in (0,1] (received {fraction})')
 
 
 class InvalidNfieldsError(StagpyError):
@@ -146,8 +145,7 @@ class InvalidNfieldsError(StagpyError):
 
     def __init__(self, nfields):
         self.nfields = nfields
-        super().__init__('nfields_max should be >5 (received {})'
-                         .format(nfields))
+        super().__init__(f'nfields_max should be >5 (received {nfields})')
 
 
 class InvalidZoomError(StagpyError):
@@ -162,8 +160,7 @@ class InvalidZoomError(StagpyError):
 
     def __init__(self, zoom):
         self.zoom = zoom
-        super().__init__('Zoom angle should be in [0,360] (received {})'
-                         .format(zoom))
+        super().__init__(f'Zoom angle should be in [0,360] (received {zoom})')
 
 
 class StagnantLidError(StagpyError):
@@ -180,7 +177,7 @@ class StagnantLidError(StagpyError):
 
     def __init__(self, sdat):
         self.sdat = sdat
-        super().__init__('Stagnant lid regime for {}'.format(sdat))
+        super().__init__(f'Stagnant lid regime for {sdat}')
 
 
 class MissingDataError(StagpyError, KeyError):

stagpy/field.py

@@ -203,7 +203,7 @@ def plot_scalar(step, var, field=None, axis=None, **extra):
         cbar = plt.colorbar(surf, cax=cax)
         cbar.set_label(meta.description +
                        (' pert.' if conf.field.perturbation else '') +
-                       (' ({})'.format(unit) if unit else ''))
+                       (f' ({unit})' if unit else ''))
     if step.geom.spherical or conf.plot.ratio is None:
         axis.set_aspect('equal')
         axis.set_axis_off()
@@ -309,8 +309,7 @@ def cmd():
                                      figsize=(9 * len(vfig), 6))
             for axis, var in zip(axes[0], vfig):
                 if var[0] not in step.fields:
-                    print("'{}' field on snap {} not found".format(var[0],
-                                                                   step.isnap))
+                    print(f"{var[0]!r} field on snap {step.isnap} not found")
                     continue
                 opts = {}
                 if var[0] in minmax:
@@ -324,7 +323,7 @@ def cmd():
             if conf.field.timelabel:
                 time, unit = sdat.scale(step.timeinfo['t'], 's')
                 time = misc.scilabel(time)
-                axes[0, 0].text(0.02, 1.02, '$t={}$ {}'.format(time, unit),
+                axes[0, 0].text(0.02, 1.02, f'$t={time}$ {unit}',
                                 transform=axes[0, 0].transAxes)
             oname = '_'.join(chain.from_iterable(vfig))
             plt.tight_layout(w_pad=3)

stagpy/misc.py

@@ -45,10 +45,10 @@ def scilabel(value, precision=2):
     Returns:
         str: the scientific notation the specified value.
     """
-    fmt = '{{:.{}e}}'.format(precision)
+    fmt = f'{{:.{precision}e}}'
     man, exp = fmt.format(value).split('e')
     exp = int(exp)
-    return r'{}\times 10^{{{}}}'.format(man, exp)
+    return fr'{man}\times 10^{{{exp}}}'
 
 
 def saveplot(fig, *name_args, close=True, **name_kwargs):
@@ -64,7 +64,7 @@ def saveplot(fig, *name_args, close=True, **name_kwargs):
         name_kwargs: keyword arguments passed on to :func:`out_name`.
     """
     oname = out_name(*name_args, **name_kwargs)
-    fig.savefig('{}.{}'.format(oname, conf.plot.format),
+    fig.savefig(f'{oname}.{conf.plot.format}',
                 format=conf.plot.format, bbox_inches='tight')
     if close:
         plt.close(fig)
@@ -160,7 +160,7 @@ class InchoateFiles:
     """
 
     def __init__(self, nfiles=1, tmp_prefix=None):
-        self._fnames = ['inchoate{}'.format(i) for i in range(nfiles)]
+        self._fnames = [f'inchoate{i}' for i in range(nfiles)]
         self._tmpprefix = tmp_prefix
         self._fids = []
 

stagpy/plates.py

@@ -480,7 +480,7 @@ def plot_plates(step, time, vrms_surface, trench, ridge, agetrench,
 
 def io_surface(timestep, time, fid, fld):
     """Output surface files."""
-    fid.write("{} {}".format(timestep, time))
+    fid.write(f"{timestep} {time}")
     fid.writelines(["%10.2e" % item for item in fld[:]])
     fid.writelines(["\n"])
 
@@ -839,4 +839,4 @@ def cmd():
             plt.plot(time, nb_plates)
             plt.subplot(122)
             plt.plot(time, ch2o)
-            misc.saveplot(figt, 'plates_{}_{}'.format(istart, iend))
+            misc.saveplot(figt, f'plates_{istart}_{iend}')

stagpy/refstate.py

@@ -17,10 +17,10 @@ def plot_ref(sdat, var):
     fig, axis = plt.subplots()
     adbts = sdat.refstate.adiabats
     if len(adbts) > 2:
-        for iad, adia in enumerate(adbts[:-1]):
+        for iad, adia in enumerate(adbts[:-1], 1):
             axis.plot(adia[var], adia['z'],
                       conf.refstate.style,
-                      label='System {}'.format(iad + 1))
+                      label=f'System {iad}')
     axis.plot(adbts[-1][var], adbts[-1]['z'],
               conf.refstate.style, color='k',
               label='Combined profile')
@@ -30,7 +30,7 @@ def plot_ref(sdat, var):
     axis.set_ylabel('z Position')
     if len(adbts) > 2:
         axis.legend()
-    misc.saveplot(fig, 'refstate_{}'.format(var))
+    misc.saveplot(fig, f'refstate_{var}')
 
 
 def cmd():

stagpy/rprof.py

@@ -36,7 +36,7 @@ def _plot_rprof_list(sdat, lovs, rprofs, stepstr):
             if xlabel:
                 _, unit = sdat.scale(1, meta.dim)
                 if unit:
-                    xlabel += ' ({})'.format(unit)
+                    xlabel += f' ({unit})'
                 axes[iplt].set_xlabel(xlabel)
             if vplt[0][:3] == 'eta':  # list of log variables
                 axes[iplt].set_xscale('log')
@@ -46,7 +46,7 @@ def _plot_rprof_list(sdat, lovs, rprofs, stepstr):
         ylabel = 'Depth' if conf.rprof.depth else 'Radius'
         _, unit = sdat.scale(1, 'm')
         if unit:
-            ylabel += ' ({})'.format(unit)
+            ylabel += f' ({unit})'
         axes[0].set_ylabel(ylabel)
         misc.saveplot(fig, fname + stepstr)
 
@@ -63,7 +63,7 @@ def plot_grid(step):
     drad, rad, _ = step.rprofs['dr']
     _, unit = step.sdat.scale(1, 'm')
     if unit:
-        unit = ' ({})'.format(unit)
+        unit = f' ({unit})'
     fig, (ax1, ax2) = plt.subplots(2, sharex=True)
     ax1.plot(rad, '-ko')
     ax1.set_ylabel('$r$' + unit)
@@ -119,7 +119,7 @@ def plot_average(sdat, lovs):
     rprof_averaged['bounds'] = (step.sdat.scale(rcmb, 'm')[0],
                                 step.sdat.scale(rsurf, 'm')[0])
 
-    stepstr = '{}_{}'.format(istart, ilast)
+    stepstr = f'{istart}_{ilast}'
 
     _plot_rprof_list(sdat, lovs, rprof_averaged, stepstr)