docstr: Tests getting a sharded shape.
docstr_tok: ["Tests","getting","a","sharded","shape","."]
code: def testGetUnpartitionedShape(self):
      """Tests getting a sharded shape."""
      p = tpu_sharding.ShardingPolicy()
      p.set_number_of_shards(3)
      p.set_shard_dimension(1)
      p.set_number_of_partitions(4)
      self.assertEqual(p.get_unpartitioned_shape([3, 5]), [3, 20])
      p.freeze()
      with self.assertRaises(ValueError):
          _ = p.get_unpartitioned_shape([3, None])
  
file_path: tensorflow/tensorflow/python/tpu/tpu_sharding_test.py
imports: from tensorflow.python.framework import tensor_shape
  from tensorflow.python.platform import test
  from tensorflow.python.tpu import tpu_sharding
label_desc: Use image node_id 6 for calling the ShardingTest obj's underlying member method code with example usage: obj.testGetUnpartitionedShape() without return types