Python Array2DExtensionType Examples

Example #1

File: test_arrow_writer.py Project: 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"))

Example #2

File: test_arrow_writer.py Project: 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())

Example #3

File: test_arrow_writer.py Project: 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")))