code: def test_generator_pass_keeps_size(
      enhancer: SpectrogramEnhancerModel, sample_input
  ):
      input_spectrograms, lengths = sample_input
      output = enhancer.forward(
          input_spectrograms=input_spectrograms, lengths=lengths
      )
  
      assert output.shape == input_spectrograms.shape
  
file_path: NeMo/tests/collections/tts/test_spectrogram_enhancer.py
imports: import pytest
  import torch
  from einops import rearrange
  from omegaconf import DictConfig
  from nemo.collections.tts.models import SpectrogramEnhancerModel
  from nemo.collections.tts.parts.utils.helpers import mask_sequence_tensor
label_desc: Use image node_id 10 for calling a global function with example usage: test_generator_pass_keeps_size(enhancer, sample_input) without return types