raise an error in case of a 'Non-fixed width type'

llama90 · May 17, 2024 · da10983 · da10983
1 parent 4baf617
commit da10983
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 1 deletion.
diff --git a/python/pyarrow/tests/test_extension_type.py b/python/pyarrow/tests/test_extension_type.py
@@ -290,7 +290,7 @@ def test_ext_type_bit_width():
     # Test for non fixed-size binary types
     ty = LabelType()
     with pytest.raises(ValueError, match="Non-fixed width type"):
-        _ = ty.byte_width
+        _ = ty.bit_width
 
 
 def test_ext_type_as_py():

diff --git a/python/pyarrow/types.pxi b/python/pyarrow/types.pxi
@@ -1524,13 +1524,17 @@ cdef class BaseExtensionType(DataType):
         """
         The byte width of the extension type.
         """
+        if self.ext_type.byte_width() == -1:
+            raise ValueError("Non-fixed width type")
         return self.ext_type.byte_width()
 
     @property
     def bit_width(self):
         """
         The bit width of the extension type.
         """
+        if self.ext_type.bit_width() == -1:
+            raise ValueError("Non-fixed width type")
         return self.ext_type.bit_width()
 
     def wrap_array(self, storage):