code: def _testVariableAxisSizePartitioner(
      self,
      name,
      axis,
      max_shard_bytes,
      expected_axis_shards,
      expected_partitions,
      max_shards=None,
  ):
      partitioner = (
          partitioned_variables.variable_axis_size_partitioner(
              axis=axis,
              max_shard_bytes=max_shard_bytes,
              max_shards=max_shards,
          )
      )
  
      with variable_scope.variable_scope(
          "root", partitioner=partitioner
      ):
          v0 = variable_scope.get_variable(
              name, dtype=dtypes.float32, shape=(4, 8, 16, 32)
          )
          v0_list = v0._get_variable_list()
          v0_part = v0._get_partitions()
          self.assertEqual(len(v0_list), expected_axis_shards)
          self.assertAllEqual(v0_part, expected_partitions)
  
imports: import os
  import numpy
  from tensorflow.python.framework import constant_op
  from tensorflow.python.framework import dtypes
  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 cond
  from tensorflow.python.ops import init_ops
  from tensorflow.python.ops import partitioned_variables
  from tensorflow.python.ops import random_ops
  from tensorflow.python.ops import variable_scope
  from tensorflow.python.ops import variable_v1
  from tensorflow.python.ops import variables
  from tensorflow.python.platform import test
  from tensorflow.python.training import saver
label_desc: Use image node_id 4 for calling the PartitionerCreatorsTest obj's underlying member method code with example usage: obj._testVariableAxisSizePartitioner(name, axis, max_shard_bytes, expected_axis_shards, expected_partitions, max_shards) without return types