.

wcfs/internal/xbtree_test.py

@@ -385,9 +385,11 @@ zbcheck._type2kind[XLOTree] = (zbcheck.TYPE_BTREE, True)
 zbcheck._btree2bucket[XLOTree] = XLOTree._bucket_type
 
 def crack_btree(ztree):
+    assert xbtree._zclassify(ztree).is_ztree, ztree
     return zbcheck.crack_btree(ztree, is_mapping=True)
 
 def crack_bucket(zbucket):
+    assert xbtree._zclassify(zbucket).is_zbucket, zbucket
     return zbcheck.crack_bucket(zbucket, is_mapping=True)
 
 
@@ -564,7 +566,7 @@ def test_restructure():
     b13 = kids[0]
     assert b13 is b1
     assert isinstance(b13, LOBucket)
-    assert crack_btree(b13) == ([1,3], [X[1], X[3]])
+    assert crack_bucket(b13) == ([1,3], [X[1], X[3]])
 
 
     # XXX tree with one committed bucket