code: def test_write_batch(self, array_feature, shape_1, shape_2):
      with tempfile.TemporaryDirectory() as tmp_dir:
          my_features = self.get_features(
              array_feature, shape_1, shape_2
          )
          dict_examples = self.get_dict_examples(shape_1, shape_2)
          dict_examples = my_features.encode_batch(dict_examples)
          with ArrowWriter(
              features=my_features,
              path=os.path.join(tmp_dir, "beta.arrow"),
          ) as writer:
              writer.write_batch(dict_examples)
              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, dict_examples["matrix"][0]
          )
          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 7 for calling the ArrayXDTest obj's underlying member method code with example usage: obj.test_write_batch(array_feature, shape_1, shape_2) without return types