docstr: Tests that the split function splits the image as expected.
docstr_tok: ["Tests","that","the","split","function","splits","the","image","as","expected","."]
code: def test_split_preset(self, image, image_high, image_low):
      """Tests that the split function splits the image as expected."""
      split = lambda image: pyramid.split(image, num_levels=1)
      image = tf.expand_dims(tf.expand_dims(image, axis=-1), axis=0)
      image_high = tf.expand_dims(
          tf.expand_dims(image_high, axis=-1), axis=0
      )
      image_low = tf.expand_dims(
          tf.expand_dims(image_low, axis=-1), axis=0
      )
  
      pyramid_split = split(image)
  
      with self.subTest(name="image_high"):
          self.assertAllClose(image_high, pyramid_split[0])
  
      with self.subTest(name="image_low"):
          self.assertAllClose(image_low, pyramid_split[1])
  
imports: from __future__ import absolute_import
  from __future__ import division
  from __future__ import print_function
  from absl.testing import parameterized
  import numpy
  from six.moves import range
  import tensorflow
  from tensorflow_graphics.image import pyramid
  from tensorflow_graphics.util import test_case
label_desc: Use image node_id 16 for calling the PyramidTest obj's underlying member method code with example usage: obj.test_split_preset(image, image_high, image_low) without return types