code: def test_mesh_creation(self):
      devices = [f"cpu:{i}" for i in range(8)]
      shape = (4, 2)
      axis_names = ["batch", "model"]
  
      mesh = distribution_lib.DeviceMesh(shape, axis_names, devices)
      self.assertEqual(mesh.shape, shape)
      self.assertEqual(mesh.axis_names, axis_names)
      self.assertEqual(mesh.devices.shape, shape)
  
imports: import os
  from unittest import mock
  import numpy
  import pytest
  import tensorflow
  from keras import backend
  from keras import testing
  from keras.backend import distribution_lib
  from keras.distribution import distribution_lib
label_desc: Use image node_id 1 for calling the DeviceMeshTest obj's underlying member method code with example usage: obj.test_mesh_creation() without return types