Update python/pyarrow/tests/test_extension_type.py

Co-authored-by: Joris Van den Bossche <[email protected]>
llama90 · May 17, 2024 · 4baf617 · 4baf617
1 parent 9f0eab8
commit 4baf617
Showing 1 changed file with 4 additions and 4 deletions.
diff --git a/python/pyarrow/tests/test_extension_type.py b/python/pyarrow/tests/test_extension_type.py
@@ -277,20 +277,20 @@ def test_ext_type_byte_width():
     # Test for non fixed-size binary types
     ty = LabelType()
     with pytest.raises(ValueError, match="Non-fixed width type"):
-        _ = ty.storage_type.byte_width
+        _ = ty.byte_width
 
 
 def test_ext_type_bit_width():
     # Test for fixed-size binary types
     ty = UuidType()
-    assert ty.storage_type.bit_width == 128
+    assert ty.bit_width == 128
     ty = ParamExtType(5)
-    assert ty.storage_type.bit_width == 40
+    assert ty.bit_width == 40
 
     # Test for non fixed-size binary types
     ty = LabelType()
     with pytest.raises(ValueError, match="Non-fixed width type"):
-        _ = ty.storage_type.byte_width
+        _ = ty.byte_width
 
 
 def test_ext_type_as_py():