Rework stylesheet reference example to cycle props

deep-jkl · deep-jkl · commit 492af3d59f98 · 2022-10-29T10:47:18.000+02:00
There is different number of circles in the last figure of each style,
when cycling is disabled.
This causes mismatch between docstring and
actual output.
diff --git a/examples/style_sheets/style_sheets_reference.py b/examples/style_sheets/style_sheets_reference.py
@@ -8,6 +8,7 @@
 line plot and histogram,
 
 """
+from itertools import cycle
 
 import numpy as np
 import matplotlib.pyplot as plt
@@ -63,7 +64,8 @@ def plot_colored_circles(ax, prng, nb_samples=15):
     the color cycle, because different styles may have different numbers
     of colors.
     """
-    for sty_dict, j in zip(plt.rcParams['axes.prop_cycle'], range(nb_samples)):
+    for sty_dict, j in zip(cycle(plt.rcParams['axes.prop_cycle']),
+                           range(nb_samples)):
         ax.add_patch(plt.Circle(prng.normal(scale=3, size=2),
                                 radius=1.0, color=sty_dict['color']))
     ax.grid(visible=True)
