code: def test_save_pretrained(model, tmp_path):
      save_directory = f"{str(tmp_path / REPO_NAME)}"
      model.save_pretrained(save_directory)
      files = os.listdir(save_directory)
      assert "model.pth" in files
      assert len(files) >= 1
  
file_path: pytorch_geometric/test/nn/test_model_hub.py
imports: import os
  from pathlib import Path
  from unittest.mock import Mock
  import pytest
  import torch
  from torch_geometric.nn import GCN
  from torch_geometric.nn.model_hub import PyGModelHubMixin
  from torch_geometric.testing import withPackage
label_desc: Use image node_id 3 for calling a global function with example usage: test_save_pretrained(model, tmp_path) without return types