Update python/pyarrow/tests/test_extension_type.py

Co-authored-by: Joris Van den Bossche <[email protected]>
llama90 · May 17, 2024 · 9f0eab8 · 9f0eab8
1 parent ac579ea
commit 9f0eab8
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/python/pyarrow/tests/test_extension_type.py b/python/pyarrow/tests/test_extension_type.py
@@ -270,9 +270,9 @@ def test_ext_type_storage_type():
 def test_ext_type_byte_width():
     # Test for fixed-size binary types
     ty = UuidType()
-    assert ty.storage_type.byte_width == 16
+    assert ty.byte_width == 16
     ty = ParamExtType(5)
-    assert ty.storage_type.byte_width == 5
+    assert ty.byte_width == 5
 
     # Test for non fixed-size binary types
     ty = LabelType()