code: def export_test(model, suffix):
      with tempfile.TemporaryDirectory() as restore_folder:
          filename = os.path.join(restore_folder, "nmt" + suffix)
          enc_filename = os.path.join(
              restore_folder, "encoder-nmt" + suffix
          )
          dec_filename = os.path.join(
              restore_folder, "decoder-nmt" + suffix
          )
          model.export(output=filename, check_trace=True)
          assert os.path.exists(enc_filename)
          assert os.path.exists(dec_filename)
  
file_path: NeMo/tests/collections/nlp/test_nmt_model.py
imports: import os
  import shutil
  import tempfile
  import pytest
  import torch
  from omegaconf import DictConfig, OmegaConf
  from nemo.collections.nlp.models import MTEncDecModel
  from nemo.collections.nlp.models.machine_translation.mt_enc_dec_config import AAYNBaseConfig
label_desc: Use image node_id 1 for calling a global function with example usage: export_test(model, suffix) without return types