Small refactoring

wafo/polynomial.py

@@ -740,19 +740,15 @@ def poly2str(p, variable='x'):
 
     def _coefstr_0(coefstr, k):
         if coefstr != '0':
-            newstr = '{0:s}'.format(coefstr)
+            return '{0:s}'.format(coefstr)
-        else:
+        return '0' if k == 0 else ''
-            newstr = '0' if k == 0 else ''
-        return newstr
 
     def _coefstr_1(coefstr, var):
         if coefstr == '0':
-            newstr = ''
+            return ''
-        elif coefstr in ['b', '1']:
+        if coefstr in ['b', '1']:
-            newstr = var
+            return var
-        else:
+        return '{0:s}*{1:s}'.format(coefstr, var)
-            newstr = '{0:s}*{1:s}'.format(coefstr, var)
-        return newstr
 
     def _coefstr_n(coefstr, var, power):
         if coefstr == '0':