66import matplotlib .pyplot as plt
77import matplotlib .category as cat
88
9- # Python2/3 text handling
10- _to_str = cat .StrCategoryFormatter ._text
11-
129
1310class TestUnitData :
1411 test_cases = [('single' , (["hello world" ], [0 ])),
@@ -156,14 +153,14 @@ def test_StrCategoryFormatter(self, ax, ydata):
156153 unit = cat .UnitData (ydata )
157154 labels = cat .StrCategoryFormatter (unit ._mapping )
158155 for i , d in enumerate (ydata ):
159- assert labels (i , i ) == _to_str ( d )
156+ assert labels (i , i ) == d
160157
161158 @pytest .mark .parametrize ("ydata" , cases , ids = ids )
162159 @pytest .mark .parametrize ("plotter" , PLOT_LIST , ids = PLOT_IDS )
163160 def test_StrCategoryFormatterPlot (self , ax , ydata , plotter ):
164161 plotter (ax , range (len (ydata )), ydata )
165162 for i , d in enumerate (ydata ):
166- assert ax .yaxis .major .formatter (i , i ) == _to_str ( d )
163+ assert ax .yaxis .major .formatter (i , i ) == d
167164 assert ax .yaxis .major .formatter (i + 1 , i + 1 ) == ""
168165 assert ax .yaxis .major .formatter (0 , None ) == ""
169166
@@ -172,7 +169,8 @@ def axis_test(axis, labels):
172169 ticks = list (range (len (labels )))
173170 np .testing .assert_array_equal (axis .get_majorticklocs (), ticks )
174171 graph_labels = [axis .major .formatter (i , i ) for i in ticks ]
175- assert graph_labels == [_to_str (l ) for l in labels ]
172+ # _text also decodes bytes as utf-8.
173+ assert graph_labels == [cat .StrCategoryFormatter ._text (l ) for l in labels ]
176174 assert list (axis .units ._mapping .keys ()) == [l for l in labels ]
177175 assert list (axis .units ._mapping .values ()) == ticks
178176
0 commit comments