docstr: Test unflattening with int and float types.
code: def test_unflatten_batch_to_2d_types(self, dtype):
      """Test unflattening with int and float types."""
      data = np.ones(shape=(6, 2), dtype=dtype)
      sizes = (2, 2, 2)
      unflattened_true = np.ones(shape=(3, 2, 2), dtype=dtype)
  
      unflattened = utils.unflatten_2d_to_batch(data, sizes)
  
      self.assertEqual(data.dtype, unflattened.dtype.as_numpy_dtype)
      self.assertAllEqual(unflattened, unflattened_true)
  
imports: from absl.testing import parameterized
  import numpy
  from scipy import linalg
  import tensorflow
  from tensorflow_graphics.geometry.convolution import utils
  from tensorflow_graphics.util import test_case
label_desc: Use image node_id 6 for calling the UtilsUnflatten2dToBatchTest obj's underlying member method code with example usage: obj.test_unflatten_batch_to_2d_types(dtype) without return types