code: def test_dvc_compression_autoencoder(shape):
      # test the compression autoencoder via run-without-exception
      x = create_input(shape)
  
      for use_gdn in (True, False):
          encoder = DVCCompressionEncoder(use_gdn=use_gdn)
          decoder = DVCCompressionDecoder(use_gdn=use_gdn)
  
          # test basic forward runs without exception
          encoded, sizes = encoder(x)
          decoded = decoder(encoded, sizes)
  
          assert list(decoded.shape) == shape
  
file_path: NeuralCompression/tests/test_models.py
imports: import numpy
  import pytest
  import torch
  from utils import create_input
  import neuralcompression.functional
  from neuralcompression.models.deep_video_compression import DVC, DVCCompressionDecoder, DVCCompressionEncoder, DVCMotionCompensationModel, DVCPyramidFlowEstimator
label_desc: Use image node_id 2 for calling a global function with example usage: test_dvc_compression_autoencoder(shape) without return types