code: def test_basic(self):
      x_all = np.random.uniform(size=(100, 32, 32, 3))
      y_all = np.random.uniform(size=(100, 1))
      sample_weight_all = None
      split = 0.2
  
      (
          x,
          y,
          sample_weight,
      ), validation_data = utils.train_validation_split(
          (x_all, y_all, sample_weight_all),
          validation_split=0.2,
          shuffle=False,
      )
  
      assert x.shape[0] == int(x_all.shape[0] * (1 - split))
      assert y.shape[0] == int(y_all.shape[0] * (1 - split))
      assert sample_weight is None
  
      (x, y, sample_weight) = validation_data
      assert x.shape[0] == int(x_all.shape[0] * split)
      assert y.shape[0] == int(y_all.shape[0] * split)
      assert sample_weight is None
  
file_path: elegy/tests/data/data_utils_test.py
imports: import math
  from unittest import TestCase
  import jax.numpy
  import numpy
  import pytest
  from elegy.data import utils
label_desc: Use image node_id 1 for calling the TrainValidationSplitTest obj's underlying member method code with example usage: obj.test_basic() without return types