code: def test_serialize_wrong_shape():
      table = Table(column_names=feature_names)
      sample = data.squeeze()
      with pytest.raises(ValueError):
          # Expected axis has 1 elements, new values have 13 elements
          table.serialize(sample)
  
      sample = data.unsqueeze(0)
      with pytest.raises(ValueError):
          # Must pass 2-d input. shape=(1, 1, 13)
          table.serialize(sample)
  
      sample = data[:, 1:]
      with pytest.raises(ValueError):
          # length mismatch
          table.serialize(sample)
  
file_path: lightning-flash/tests/core/serve/test_types/test_table.py
imports: import pytest
  import torch
  from flash.core.serve.types import Table
  from flash.core.utilities.imports import _TOPIC_SERVE_AVAILABLE
label_desc: Use image node_id 2 for calling a global function with example usage: test_serialize_wrong_shape() without return types