code: def test_write(self, array_feature, shape_1, shape_2):
      with tempfile.TemporaryDirectory() as tmp_dir:
          my_features = self.get_features(
              array_feature, shape_1, shape_2
          )
          my_examples = [
              (0, self.get_dict_example_0(shape_1, shape_2)),
              (1, self.get_dict_example_1(shape_1, shape_2)),
          ]
          with ArrowWriter(
              features=my_features,
              path=os.path.join(tmp_dir, "beta.arrow"),
          ) as writer:
              for key, record in my_examples:
                  example = my_features.encode_example(record)
                  writer.write(example)
              num_examples, num_bytes = writer.finalize()
          dataset = datasets.Dataset.from_file(
              os.path.join(tmp_dir, "beta.arrow")
          )
          self._check_getitem_output_type(
              dataset, shape_1, shape_2, my_examples[0][1]["matrix"]
          )
          del dataset
  
imports: import os
  import random
  import tempfile
  import unittest
  import numpy
  import pandas
  import pyarrow
  import pytest
  from absl.testing import parameterized
  import datasets
  from datasets.arrow_writer import ArrowWriter
  from datasets.features import Array2D, Array3D, Array4D, Array5D, Value
  from datasets.features.features import Array3DExtensionType, PandasArrayExtensionDtype, _ArrayXD
  from datasets.formatting.formatting import NumpyArrowExtractor, SimpleArrowExtractor
label_desc: Use image node_id 6 for calling the ArrayXDTest obj's underlying member method code with example usage: obj.test_write(array_feature, shape_1, shape_2) without return types