File tree Expand file tree Collapse file tree 2 files changed +14
-2
lines changed Expand file tree Collapse file tree 2 files changed +14
-2
lines changed Original file line number Diff line number Diff line change @@ -399,10 +399,14 @@ def __eq__(self, other: Any) -> bool:
399
399
400
400
def __repr__ (self ) -> str_type :
401
401
if self .categories is None :
402
- data = "None, "
402
+ data = "None"
403
403
else :
404
404
data = self .categories ._format_data (name = type (self ).__name__ )
405
- return f"CategoricalDtype(categories={ data } ordered={ self .ordered } )"
405
+ if data is None :
406
+ # self.categories is RangeIndex
407
+ data = str (self .categories ._range )
408
+ data = data .rstrip (", " )
409
+ return f"CategoricalDtype(categories={ data } , ordered={ self .ordered } )"
406
410
407
411
@staticmethod
408
412
def _hash_categories (categories , ordered : Ordered = True ) -> int :
Original file line number Diff line number Diff line change @@ -199,6 +199,14 @@ def test_not_string(self):
199
199
# though CategoricalDtype has object kind, it cannot be string
200
200
assert not is_string_dtype (CategoricalDtype ())
201
201
202
+ def test_repr_range_categories (self ):
203
+ rng = pd .Index (range (3 ))
204
+ dtype = CategoricalDtype (categories = rng , ordered = False )
205
+ result = repr (dtype )
206
+
207
+ expected = "CategoricalDtype(categories=range(0, 3), ordered=False)"
208
+ assert result == expected
209
+
202
210
203
211
class TestDatetimeTZDtype (Base ):
204
212
@pytest .fixture
You can’t perform that action at this time.
0 commit comments