code: def test_unsqueeze_expand(input_shape, dim, size, output_shape):
      a = torch.randn(size=input_shape)
      b = unsqueeze_expand(a, dim=dim, size=size)
      assert b.shape == output_shape
  
imports: from typing import List
  import pytest
  import torch
  from gluonts.torch.util import lagged_sequence_values, unsqueeze_expand
label_desc: Use image node_id 2 for calling a global function with example usage: test_unsqueeze_expand(input_shape, dim, size, output_shape) without return types