Simplify series plotting test

pandas-dev · Nov 20, 2017 · ce932a6 · ce932a6
1 parent 9de60c1
commit ce932a6
Showing 1 changed file with 2 additions and 3 deletions.
diff --git a/pandas/tests/plotting/test_series.py b/pandas/tests/plotting/test_series.py
@@ -3,7 +3,7 @@
 """ Test cases for Series.plot """
 
 
-import itertools
+from itertools import chain
 import pytest
 
 from datetime import datetime
@@ -333,8 +333,7 @@ def test_pie_series(self):
                                autopct='%.2f', fontsize=7)
         pcts = ['{0:.2f}'.format(s * 100)
                 for s in series.values / float(series.sum())]
-        iters = [iter(series.index), iter(pcts)]
-        expected_texts = [next(it) for it in itertools.cycle(iters)]
+        expected_texts = list(chain.from_iterable(zip(series.index, pcts)))
         self._check_text_labels(ax.texts, expected_texts)
         for t in ax.texts:
             assert t.get_fontsize() == 7