code: def test_split_data_has_one_batch_error():
      with pytest.raises(ValueError) as info:
          data_utils.split_dataset(
              tf.data.Dataset.from_tensor_slices(
                  np.array([1, 2, 3])
              ).batch(12),
              0.2,
          )
  
      assert "The dataset should at least contain 2 batches" in str(
          info.value
      )
  
file_path: autokeras/autokeras/utils/data_utils_test.py
imports: import numpy
  import pytest
  import tensorflow
  from autokeras.utils import data_utils
label_desc: Use image node_id 1 for calling a global function with example usage: test_split_data_has_one_batch_error() without return types