Python Array2DExtensionType示例

示例#1

文件： test_arrow_writer.py 项目： ml-lab/datasets-1

 def test_try_compatible_extension_type(self):
     arr = pa.array(
         TypedSequence([[[1, 2, 3]]],
                       try_type=Array2DExtensionType((1, 3), "int64")))
     self.assertEqual(arr.type, Array2DExtensionType((1, 3), "int64"))

示例#2

文件： test_arrow_writer.py 项目： ml-lab/datasets-1

 def test_try_incompatible_extension_type(self):
     arr = pa.array(
         TypedSequence(["foo", "bar"],
                       try_type=Array2DExtensionType((1, 3), "int64")))
     self.assertEqual(arr.type, pa.string())

示例#3

文件： test_arrow_writer.py 项目： ml-lab/datasets-1

 def test_incompatible_extension_type(self):
     with self.assertRaises((TypeError, pa.lib.ArrowInvalid)):
         _ = pa.array(
             TypedSequence(["foo", "bar"],
                           type=Array2DExtensionType((1, 3), "int64")))