code: def test_basic(self):
      x = np.random.uniform(size=(100, 32, 32, 3))
      y = np.random.uniform(size=(100, 1))
      batch_size = 10
      epochs = 1
      data_adapter = ArrayDataAdapter(
          x,
          y=y,
          sample_weights=None,
          batch_size=batch_size,
          epochs=epochs,
          steps=None,
          shuffle=False,
      )
      num_steps = math.ceil(x.shape[0] / batch_size) * epochs
      iterator_fn = data_adapter.get_dataset()
      for i, batch in zip(range(num_steps), iterator_fn()):
          batch_x, batch_y = batch
          assert batch_x.shape == (batch_size, *x.shape[1:])
          assert batch_y.shape == (batch_size, *y.shape[1:])
          np.testing.assert_array_equal(
              batch_x, x[i * batch_size : (i + 1) * batch_size]
          )
  
      data_adapter.get_size() == x.shape[0]
      data_adapter.partial_batch_size == 0
  
file_path: elegy/tests/data/array_adapter_test.py
imports: import math
  from unittest import TestCase
  import jax.numpy
  import numpy
  import pytest
  from elegy.data.array_adapter import ArrayDataAdapter
label_desc: Use image node_id 1 for calling the ArrayDataAdapterTest obj's underlying member method code with example usage: obj.test_basic() without return types