code: def testVariableShapeFunction(self):
      # size_splits too big
      with self.assertRaises(ValueError):
          array_ops.split([0, 1], [3, -1], axis=0)
  
      # Correct inference of variable dimension
      s0, s1 = array_ops.split([0, 1, 2], [2, -1], axis=0)
      assert s0.shape.as_list() == [2]
      assert s1.shape.as_list() == [1]
  
imports: import numpy
  from tensorflow.python.framework import constant_op
  from tensorflow.python.framework import dtypes
  from tensorflow.python.framework import errors_impl
  from tensorflow.python.framework import ops
  from tensorflow.python.framework import test_util
  from tensorflow.python.ops import array_ops
  from tensorflow.python.ops import gradients_impl
  from tensorflow.python.ops import math_ops
  from tensorflow.python.platform import test
label_desc: Use image node_id 25 for calling the SplitOpTest obj's underlying member method code with example usage: obj.testVariableShapeFunction() without return types