code: def test_shapes(param):
      model, out_size = param
      model = model(weights="DEFAULT")
      model.eval()
      output_batch = model(INPUT_BATCH)
      assert output_batch.size() == (BATCH_SIZE, out_size)
      assert model.output_size() == out_size
  
file_path: pykale/tests/embed/test_image_cnn.py
imports: import pytest
  import torch
  from kale.embed.image_cnn import Flatten, Identity, LeNet, ResNet18Feature, ResNet34Feature, ResNet50Feature, ResNet101Feature, ResNet152Feature, SimpleCNNBuilder, SmallCNNFeature
label_desc: Use image node_id 3 for calling a global function with example usage: test_shapes(param) without return types