code: def test_squeeze():
      data = np.array([[[0], [1], [2]]])
      x = tensor(data)
  
      t = squeeze(x)
      assert t.shape == (3,)
      assert t.dtype is not None
  
      t = squeeze(x, axis=0)
      assert t.shape == (3, 1)
  
      with pytest.raises(ValueError):
          squeeze(x, axis=1)
  
      t = squeeze(x, axis=2)
      assert t.shape == (1, 3)
  
imports: import numpy
  import pytest
  from ....core import tile
  from ....core.operand import OperandStage
  from ...datasource import ones, tensor, arange
  from ..None import transpose, broadcast_to, where, argwhere, array_split, split, squeeze, result_type, repeat, copyto, isin, moveaxis, TensorCopyTo, atleast_1d, atleast_2d, atleast_3d, ravel, searchsorted, unique, sort, partition, topk, to_gpu, to_cpu
label_desc: Use image node_id 12 for calling a global function with example usage: test_squeeze() without return types