code: def test_logical_to_mesh_axes_priorities(self):
      p_rules = (("foo", "model"), ("bar", "model"), ("baz", "data"))
      with partitioning.axis_rules(p_rules):
          self.assertEqual(
              partitioning.logical_to_mesh_axes(("foo", "bar", "baz")),
              ("model", None, "data"),
          )
          self.assertEqual(
              partitioning.logical_to_mesh_axes(("bar", "foo", "baz")),
              (None, "model", "data"),
          )
          self.assertEqual(
              partitioning.logical_to_mesh_axes(("baz", "bar", "foo")),
              ("data", None, "model"),
          )
          self.assertEqual(
              partitioning.logical_to_mesh_axes(
                  ("baz", "bar", "foo", "unassigned")
              ),
              ("data", None, "model", None),
          )
  
imports: import jax
  import jax.numpy
  from absl.testing import absltest, parameterized
  from jax import random, sharding
  from jax.experimental import mesh_utils
  from flax import linen
  from flax.core import freeze, unfreeze
  from flax.linen import partitioning
label_desc: Use image node_id 4 for calling the PartitioningTest obj's underlying member method code with example usage: obj.test_logical_to_mesh_axes_priorities() without return types