@@ -540,6 +540,13 @@ def _compare(old_mgr, new_mgr):
540
540
assert new_mgr .get ("g" ).dtype == np .float64
541
541
assert new_mgr .get ("h" ).dtype == np .float16
542
542
543
+ def test_invalid_ea_block (self ):
544
+ with pytest .raises (AssertionError , match = "block.size != values.size" ):
545
+ create_mgr ("a: category; b: category" )
546
+
547
+ with pytest .raises (AssertionError , match = "block.size != values.size" ):
548
+ create_mgr ("a: category2; b: category2" )
549
+
543
550
def test_interleave (self ):
544
551
545
552
# self
@@ -552,14 +559,10 @@ def test_interleave(self):
552
559
# will be converted according the actual dtype of the underlying
553
560
mgr = create_mgr ("a: category" )
554
561
assert mgr .as_array ().dtype == "i8"
555
- mgr = create_mgr ("a: category; b: category" )
556
- assert mgr .as_array ().dtype == "i8"
557
562
mgr = create_mgr ("a: category; b: category2" )
558
563
assert mgr .as_array ().dtype == "object"
559
564
mgr = create_mgr ("a: category2" )
560
565
assert mgr .as_array ().dtype == "object"
561
- mgr = create_mgr ("a: category2; b: category2" )
562
- assert mgr .as_array ().dtype == "object"
563
566
564
567
# combinations
565
568
mgr = create_mgr ("a: f8" )
@@ -702,7 +705,7 @@ def test_equals(self):
702
705
"a:i8;b:f8" , # basic case
703
706
"a:i8;b:f8;c:c8;d:b" , # many types
704
707
"a:i8;e:dt;f:td;g:string" , # more types
705
- "a:i8;b:category;c:category2;d:category2 " , # categories
708
+ "a:i8;b:category;c:category2" , # categories
706
709
"c:sparse;d:sparse_na;b:f8" , # sparse
707
710
],
708
711
)
0 commit comments