Use context for switching styles

rougier · rougier · commit 7b59d6104a2a · 2020-07-08T08:02:55.000+02:00
diff --git a/scripts/styles.py b/scripts/styles.py
@@ -6,19 +6,16 @@
 import matplotlib as mpl
 import matplotlib.pyplot as plt
 
-styles = ['default'] + plt.style.available
-
-for style in styles:
-    fig = plt.figure(figsize=(5,3), dpi=100)
-    plt.style.use(style)
-    ax = plt.subplot(1,1,1)
-    X = np.linspace(0,2*np.pi, 256)
-    Y = np.cos(X)
-    ax.plot(X,Y)
-    plt.title(style, family="Source Serif Pro", size=32)
-    plt.tight_layout()
-    # plt.show()
-    plt.savefig("../figures/style-%s.pdf" % style)
-    plt.close(fig)
+for style in ['default'] + plt.style.available:
+    with plt.style.context(style):
+        fig = plt.figure(figsize=(5,3), dpi=100)
+        ax = plt.subplot(1,1,1)
+        X = np.linspace(0,2*np.pi, 256)
+        Y = np.cos(X)
+        ax.plot(X,Y)
+        plt.title(style, family="Source Serif Pro", size=32)
+        plt.tight_layout()
+        plt.savefig("../figures/style-%s.pdf" % style)
+        plt.close(fig)
 
 
