@@ -39,62 +39,71 @@ def test_creation(c):
39
39
_cycler_helper (c , 3 , ['c' ], [['r' , 'g' , 'b' ]])
40
40
41
41
42
- def test_compose ():
42
+ def test_add ():
43
43
c1 = cycler (c = 'rgb' )
44
44
c2 = cycler (lw = range (3 ))
45
- c3 = cycler (lw = range (15 ))
46
45
# addition
47
- yield _cycler_helper , c1 + c2 , 3 , ['c' , 'lw' ], [list ('rgb' ), range (3 )]
48
- yield _cycler_helper , c2 + c1 , 3 , ['c' , 'lw' ], [list ('rgb' ), range (3 )]
49
- yield _cycles_equal , c2 + c1 , c1 + c2
46
+ _cycler_helper (c1 + c2 , 3 , ['c' , 'lw' ], [list ('rgb' ), range (3 )])
47
+ _cycler_helper (c2 + c1 , 3 , ['c' , 'lw' ], [list ('rgb' ), range (3 )])
48
+ _cycles_equal (c2 + c1 , c1 + c2 )
49
+
50
+
51
+ def test_add_len_mismatch ():
50
52
# miss-matched add lengths
53
+ c1 = cycler (c = 'rgb' )
54
+ c3 = cycler (lw = range (15 ))
51
55
with pytest .raises (ValueError ):
52
56
c1 + c3
53
57
with pytest .raises (ValueError ):
54
58
c3 + c1
55
59
60
+
61
+ def test_prod ():
62
+ c1 = cycler (c = 'rgb' )
63
+ c2 = cycler (lw = range (3 ))
64
+ c3 = cycler (lw = range (15 ))
56
65
# multiplication
57
66
target = zip (* product (list ('rgb' ), range (3 )))
58
- yield ( _cycler_helper , c1 * c2 , 9 , ['c' , 'lw' ], target )
67
+ _cycler_helper ( c1 * c2 , 9 , ['c' , 'lw' ], target )
59
68
60
69
target = zip (* product (range (3 ), list ('rgb' )))
61
- yield ( _cycler_helper , c2 * c1 , 9 , ['lw' , 'c' ], target )
70
+ _cycler_helper ( c2 * c1 , 9 , ['lw' , 'c' ], target )
62
71
63
72
target = zip (* product (range (15 ), list ('rgb' )))
64
- yield ( _cycler_helper , c3 * c1 , 45 , ['lw' , 'c' ], target )
73
+ _cycler_helper ( c3 * c1 , 45 , ['lw' , 'c' ], target )
65
74
66
75
67
76
def test_inplace ():
68
77
c1 = cycler (c = 'rgb' )
69
78
c2 = cycler (lw = range (3 ))
70
79
c2 += c1
71
- yield _cycler_helper , c2 , 3 , ['c' , 'lw' ], [list ('rgb' ), range (3 )]
80
+ _cycler_helper ( c2 , 3 , ['c' , 'lw' ], [list ('rgb' ), range (3 )])
72
81
73
82
c3 = cycler (c = 'rgb' )
74
83
c4 = cycler (lw = range (3 ))
75
84
c3 *= c4
76
85
target = zip (* product (list ('rgb' ), range (3 )))
77
- yield ( _cycler_helper , c3 , 9 , ['c' , 'lw' ], target )
86
+ _cycler_helper ( c3 , 9 , ['c' , 'lw' ], target )
78
87
79
88
80
89
def test_constructor ():
81
90
c1 = cycler (c = 'rgb' )
82
91
c2 = cycler (ec = c1 )
83
- yield _cycler_helper , c1 + c2 , 3 , ['c' , 'ec' ], [['r' , 'g' , 'b' ]]* 2
92
+ _cycler_helper ( c1 + c2 , 3 , ['c' , 'ec' ], [['r' , 'g' , 'b' ]]* 2 )
84
93
c3 = cycler (c = c1 )
85
- yield _cycler_helper , c3 + c2 , 3 , ['c' , 'ec' ], [['r' , 'g' , 'b' ]]* 2
94
+ _cycler_helper ( c3 + c2 , 3 , ['c' , 'ec' ], [['r' , 'g' , 'b' ]]* 2 )
86
95
# Using a non-string hashable
87
96
c4 = cycler (1 , range (3 ))
88
- yield _cycler_helper , c4 + c1 , 3 , [1 , 'c' ], [range (3 ), ['r' , 'g' , 'b' ]]
97
+ _cycler_helper ( c4 + c1 , 3 , [1 , 'c' ], [range (3 ), ['r' , 'g' , 'b' ]])
89
98
90
99
# addition using cycler()
91
- yield ( _cycler_helper , cycler (c = 'rgb' , lw = range (3 )),
92
- 3 , ['c' , 'lw' ], [list ('rgb' ), range (3 )])
93
- yield ( _cycler_helper , cycler (lw = range (3 ), c = 'rgb' ),
94
- 3 , ['c' , 'lw' ], [list ('rgb' ), range (3 )])
100
+ _cycler_helper ( cycler (c = 'rgb' , lw = range (3 )),
101
+ 3 , ['c' , 'lw' ], [list ('rgb' ), range (3 )])
102
+ _cycler_helper ( cycler (lw = range (3 ), c = 'rgb' ),
103
+ 3 , ['c' , 'lw' ], [list ('rgb' ), range (3 )])
95
104
# Purposely mixing them
96
- yield ( _cycler_helper , cycler (c = range (3 ), lw = c1 ),
97
- 3 , ['c' , 'lw' ], [range (3 ), list ('rgb' )])
105
+ _cycler_helper ( cycler (c = range (3 ), lw = c1 ),
106
+ 3 , ['c' , 'lw' ], [range (3 ), list ('rgb' )])
98
107
99
108
100
109
def test_failures ():
@@ -116,24 +125,24 @@ def test_simplify():
116
125
c1 = cycler (c = 'rgb' )
117
126
c2 = cycler (ec = c1 )
118
127
for c in [c1 * c2 , c2 * c1 , c1 + c2 ]:
119
- yield _cycles_equal , c , c .simplify ()
128
+ _cycles_equal ( c , c .simplify () )
120
129
121
130
122
131
def test_multiply ():
123
132
c1 = cycler (c = 'rgb' )
124
- yield _cycler_helper , 2 * c1 , 6 , ['c' ], ['rgb' * 2 ]
133
+ _cycler_helper ( 2 * c1 , 6 , ['c' ], ['rgb' * 2 ])
125
134
126
135
c2 = cycler (ec = c1 )
127
136
c3 = c1 * c2
128
137
129
- yield _cycles_equal , 2 * c3 , c3 * 2
138
+ _cycles_equal ( 2 * c3 , c3 * 2 )
130
139
131
140
132
141
def test_mul_fails ():
133
142
c1 = cycler (c = 'rgb' )
134
- pytest .raises (TypeError , mul , c1 , 2.0 )
135
- pytest .raises (TypeError , mul , c1 , 'a' )
136
- pytest .raises (TypeError , mul , c1 , [])
143
+ pytest .raises (TypeError , mul , c1 , 2.0 )
144
+ pytest .raises (TypeError , mul , c1 , 'a' )
145
+ pytest .raises (TypeError , mul , c1 , [])
137
146
138
147
139
148
def test_getitem ():
@@ -143,7 +152,7 @@ def test_getitem():
143
152
slice (None , None , - 1 ),
144
153
slice (1 , 5 , None ),
145
154
slice (0 , 5 , 2 )):
146
- yield _cycles_equal , c1 [slc ], cycler (3 , widths [slc ])
155
+ _cycles_equal ( c1 [slc ], cycler (3 , widths [slc ]) )
147
156
148
157
149
158
def test_fail_getime ():
@@ -166,14 +175,14 @@ def test_repr():
166
175
c_sum_rpr = "(cycler('c', ['r', 'g', 'b']) + cycler('3rd', [0, 1, 2]))"
167
176
c_prod_rpr = "(cycler('c', ['r', 'g', 'b']) * cycler('3rd', [0, 1, 2]))"
168
177
169
- yield _repr_tester_helper , '__repr__' , c + c2 , c_sum_rpr
170
- yield _repr_tester_helper , '__repr__' , c * c2 , c_prod_rpr
178
+ _repr_tester_helper ( '__repr__' , c + c2 , c_sum_rpr )
179
+ _repr_tester_helper ( '__repr__' , c * c2 , c_prod_rpr )
171
180
172
181
sum_html = "<table><th>'3rd'</th><th>'c'</th><tr><td>0</td><td>'r'</td></tr><tr><td>1</td><td>'g'</td></tr><tr><td>2</td><td>'b'</td></tr></table>"
173
182
prod_html = "<table><th>'3rd'</th><th>'c'</th><tr><td>0</td><td>'r'</td></tr><tr><td>1</td><td>'r'</td></tr><tr><td>2</td><td>'r'</td></tr><tr><td>0</td><td>'g'</td></tr><tr><td>1</td><td>'g'</td></tr><tr><td>2</td><td>'g'</td></tr><tr><td>0</td><td>'b'</td></tr><tr><td>1</td><td>'b'</td></tr><tr><td>2</td><td>'b'</td></tr></table>"
174
183
175
- yield _repr_tester_helper , '_repr_html_' , c + c2 , sum_html
176
- yield _repr_tester_helper , '_repr_html_' , c * c2 , prod_html
184
+ _repr_tester_helper ( '_repr_html_' , c + c2 , sum_html )
185
+ _repr_tester_helper ( '_repr_html_' , c * c2 , prod_html )
177
186
178
187
179
188
def test_call ():
@@ -266,17 +275,17 @@ def _eq_test_helper(a, b, res):
266
275
def test_eq ():
267
276
a = cycler (c = 'rgb' )
268
277
b = cycler (c = 'rgb' )
269
- yield _eq_test_helper , a , b , True
270
- yield _eq_test_helper , a , b [::- 1 ], False
278
+ _eq_test_helper ( a , b , True )
279
+ _eq_test_helper ( a , b [::- 1 ], False )
271
280
c = cycler (lw = range (3 ))
272
- yield _eq_test_helper , a + c , c + a , True
273
- yield _eq_test_helper , a + c , c + b , True
274
- yield _eq_test_helper , a * c , c * a , False
275
- yield _eq_test_helper , a , c , False
281
+ _eq_test_helper ( a + c , c + a , True )
282
+ _eq_test_helper ( a + c , c + b , True )
283
+ _eq_test_helper ( a * c , c * a , False )
284
+ _eq_test_helper ( a , c , False )
276
285
d = cycler (c = 'ymk' )
277
- yield _eq_test_helper , b , d , False
286
+ _eq_test_helper ( b , d , False )
278
287
e = cycler (c = 'orange' )
279
- yield _eq_test_helper , b , e , False
288
+ _eq_test_helper ( b , e , False )
280
289
281
290
282
291
def test_cycler_exceptions ():
@@ -325,15 +334,15 @@ def test_by_key_add():
325
334
cy = cycler (c = input_dict ['c' ]) + cycler (lw = input_dict ['lw' ])
326
335
res = cy .by_key ()
327
336
assert res == input_dict
328
- yield _by_key_helper , cy
337
+ _by_key_helper ( cy )
329
338
330
339
331
340
def test_by_key_mul ():
332
341
input_dict = dict (c = list ('rg' ), lw = [1 , 2 , 3 ])
333
342
cy = cycler (c = input_dict ['c' ]) * cycler (lw = input_dict ['lw' ])
334
343
res = cy .by_key ()
335
344
assert input_dict ['lw' ] * len (input_dict ['c' ]) == res ['lw' ]
336
- yield _by_key_helper , cy
345
+ _by_key_helper ( cy )
337
346
338
347
339
348
def test_contains ():
0 commit comments