pyupgrade

Author: nschloe

test/test_basic_sin.py

@@ -6,7 +6,7 @@
 
 def plot():
     fig = plt.figure()
-    with plt.style.context(("ggplot")):
+    with plt.style.context("ggplot"):
         t = np.arange(0.0, 2.0, 0.1)
         s = np.sin(2 * np.pi * t)
         s2 = np.cos(2 * np.pi * t)

test/test_errorband.py

@@ -6,7 +6,7 @@
 
 def plot():
     fig, ax = plt.subplots()
-    with plt.style.context(("ggplot")):
+    with plt.style.context("ggplot"):
         t = np.linspace(0, 2 * np.pi, 11)
         s = np.sin(t)
         ax.plot(t, s, "k-")

test/test_externalize_tables.py

@@ -6,7 +6,7 @@ def plot():
     import numpy as np
 
     fig = plt.figure()
-    with plt.style.context(("ggplot")):
+    with plt.style.context("ggplot"):
         t = np.arange(0.0, 2.0, 0.1)
         s = np.sin(2 * np.pi * t)
         s2 = np.cos(2 * np.pi * t)

test/test_line_color_marker.py

@@ -6,7 +6,7 @@ def plot():
     import numpy as np
 
     fig = plt.figure()
-    with plt.style.context(("ggplot")):
+    with plt.style.context("ggplot"):
         t = np.arange(0.0, 2.0, 0.1)
         s = np.sin(2 * np.pi * t)
         s2 = np.cos(2 * np.pi * t)

test/test_marker.py

@@ -6,7 +6,7 @@
 
 def plot():
     fig, ax = plt.subplots()
-    with plt.style.context(("ggplot")):
+    with plt.style.context("ggplot"):
         t = np.linspace(0, 2 * np.pi, 11)
         s = np.sin(t)
         c = np.cos(t)

test/test_scatter.py

@@ -6,7 +6,7 @@ def plot():
     import numpy as np
 
     fig = plt.figure()
-    with plt.style.context(("fivethirtyeight")):
+    with plt.style.context("fivethirtyeight"):
         np.random.seed(123)
         plt.scatter(
             np.linspace(0, 100, 101),

tikzplotlib/_axes.py

@@ -44,26 +44,26 @@ def __init__(self, data, obj):  # noqa: C901
         data["current axis title"] = title
         if title:
             title = _common_texification(title)
-            self.axis_options.append(u"title={{{}}}".format(title))
+            self.axis_options.append("title={{{}}}".format(title))
 
         # get axes titles
         xlabel = obj.get_xlabel()
         xrotation = obj.xaxis.get_label().get_rotation()
         if xlabel:
             xlabel = _common_texification(xlabel)
-            self.axis_options.append(u"xlabel={{{}}}".format(xlabel))
+            self.axis_options.append("xlabel={{{}}}".format(xlabel))
             if xrotation != 0:
                 self.axis_options.append(
-                    u"xlabel style={{rotate={}}}".format(xrotation - 90)
+                    "xlabel style={{rotate={}}}".format(xrotation - 90)
                 )
         ylabel = obj.get_ylabel()
         yrotation = obj.yaxis.get_label().get_rotation()
         if ylabel:
             ylabel = _common_texification(ylabel)
-            self.axis_options.append(u"ylabel={{{}}}".format(ylabel))
+            self.axis_options.append("ylabel={{{}}}".format(ylabel))
             if yrotation != 90:
                 self.axis_options.append(
-                    u"ylabel style={{rotate={}}}".format(yrotation - 90)
+                    "ylabel style={{rotate={}}}".format(yrotation - 90)
                 )
 
         # Axes limits.
@@ -573,7 +573,7 @@ def _get_ticks(data, xy, ticks, ticklabels):
         # must appear in the TikZ plot.
         if label:
             try:
-                label_float = float(label.replace(u"\N{MINUS SIGN}", "-"))
+                label_float = float(label.replace("\N{MINUS SIGN}", "-"))
                 is_label_required = is_label_required or (label and label_float != tick)
             except ValueError:
                 is_label_required = True
@@ -597,7 +597,7 @@ def _get_ticks(data, xy, ticks, ticklabels):
 
         if is_label_required:
             axis_options.append(
-                u"{}ticklabels={{{}}}".format(xy, u",".join(pgfplots_ticklabels))
+                "{}ticklabels={{{}}}".format(xy, ",".join(pgfplots_ticklabels))
             )
     return axis_options
 

tikzplotlib/_line2d.py

@@ -201,7 +201,7 @@ def _table(obj, data):  # noqa: C901
     elif isinstance(xdata_alt[0], str):
         data["current axes"].axis_options += [
             "xtick={{{}}}".format(",".join([ff.format(x) for x in xdata])),
-            u"xticklabels={{{}}}".format(u",".join(xdata_alt)),
+            "xticklabels={{{}}}".format(",".join(xdata_alt)),
         ]
         xdata, ydata = transform_to_data_coordinates(obj, xdata, ydata)
     else:

tikzplotlib/_save.py

@@ -222,15 +222,14 @@ def get_tikz_code(
         code += "\\end{tikzpicture}\n"
 
     if standalone:
-        # create a latex wrapper for the tikz
-        # <https://tex.stackexchange.com/a/361070/13262>
+        # When using pdflatex, this is necessary:
+        # \\DeclareUnicodeCharacter{{2212}}{{−}}
         code = """\\documentclass{{standalone}}
 \\usepackage[utf8]{{inputenc}}
 \\usepackage{{pgfplots}}
 \\usepgfplotslibrary{{groupplots}}
 \\usepgfplotslibrary{{dateplot}}
 \\pgfplotsset{{compat=newest}}
-\\DeclareUnicodeCharacter{{2212}}{{−}}
 \\begin{{document}}
 {}
 \\end{{document}}\n""".format(