@@ -277,21 +277,19 @@ def test_constructor_with_index(self):
277
277
def test_constructor_with_generator (self ):
278
278
# This was raising an Error in isna(single_val).any() because isna
279
279
# returned a scalar for a generator
280
- xrange = range
281
280
282
281
exp = Categorical ([0 , 1 , 2 ])
283
282
cat = Categorical ((x for x in [0 , 1 , 2 ]))
284
283
tm .assert_categorical_equal (cat , exp )
285
- cat = Categorical (xrange (3 ))
284
+ cat = Categorical (range (3 ))
286
285
tm .assert_categorical_equal (cat , exp )
287
286
288
- # This uses xrange internally
289
287
MultiIndex .from_product ([range (5 ), ["a" , "b" , "c" ]])
290
288
291
289
# check that categories accept generators and sequences
292
290
cat = Categorical ([0 , 1 , 2 ], categories = (x for x in [0 , 1 , 2 ]))
293
291
tm .assert_categorical_equal (cat , exp )
294
- cat = Categorical ([0 , 1 , 2 ], categories = xrange (3 ))
292
+ cat = Categorical ([0 , 1 , 2 ], categories = range (3 ))
295
293
tm .assert_categorical_equal (cat , exp )
296
294
297
295
@pytest .mark .parametrize (
0 commit comments