code: def test_get_model_files_only_pt(self):
      # BERT
      bert_files = get_model_files("bert", frameworks=["pt"])
  
      doc_file = str(
          Path(bert_files["doc_file"]).relative_to(REPO_PATH)
      )
      self.assertEqual(doc_file, "docs/source/en/model_doc/bert.md")
  
      model_files = {
          str(Path(f).relative_to(REPO_PATH))
          for f in bert_files["model_files"]
      }
      bert_model_files = BERT_MODEL_FILES - {
          "src/transformers/models/bert/modeling_tf_bert.py",
          "src/transformers/models/bert/modeling_flax_bert.py",
      }
      self.assertEqual(model_files, bert_model_files)
  
      self.assertEqual(bert_files["module_name"], "bert")
  
      test_files = {
          str(Path(f).relative_to(REPO_PATH))
          for f in bert_files["test_files"]
      }
      bert_test_files = {
          "tests/models/bert/test_tokenization_bert.py",
          "tests/models/bert/test_modeling_bert.py",
      }
      self.assertEqual(test_files, bert_test_files)
  
      # VIT
      vit_files = get_model_files("vit", frameworks=["pt"])
      doc_file = str(Path(vit_files["doc_file"]).relative_to(REPO_PATH))
      self.assertEqual(doc_file, "docs/source/en/model_doc/vit.md")
  
      model_files = {
          str(Path(f).relative_to(REPO_PATH))
          for f in vit_files["model_files"]
      }
      vit_model_files = VIT_MODEL_FILES - {
          "src/transformers/models/vit/modeling_tf_vit.py",
          "src/transformers/models/vit/modeling_flax_vit.py",
      }
      self.assertEqual(model_files, vit_model_files)
  
      self.assertEqual(vit_files["module_name"], "vit")
  
      test_files = {
          str(Path(f).relative_to(REPO_PATH))
          for f in vit_files["test_files"]
      }
      vit_test_files = {
          "tests/models/vit/test_image_processing_vit.py",
          "tests/models/vit/test_modeling_vit.py",
      }
      self.assertEqual(test_files, vit_test_files)
  
      # Wav2Vec2
      wav2vec2_files = get_model_files("wav2vec2", frameworks=["pt"])
      doc_file = str(
          Path(wav2vec2_files["doc_file"]).relative_to(REPO_PATH)
      )
      self.assertEqual(doc_file, "docs/source/en/model_doc/wav2vec2.md")
  
      model_files = {
          str(Path(f).relative_to(REPO_PATH))
          for f in wav2vec2_files["model_files"]
      }
      wav2vec2_model_files = WAV2VEC2_MODEL_FILES - {
          "src/transformers/models/wav2vec2/modeling_tf_wav2vec2.py",
          "src/transformers/models/wav2vec2/modeling_flax_wav2vec2.py",
      }
      self.assertEqual(model_files, wav2vec2_model_files)
  
      self.assertEqual(wav2vec2_files["module_name"], "wav2vec2")
  
      test_files = {
          str(Path(f).relative_to(REPO_PATH))
          for f in wav2vec2_files["test_files"]
      }
      wav2vec2_test_files = {
          "tests/models/wav2vec2/test_feature_extraction_wav2vec2.py",
          "tests/models/wav2vec2/test_modeling_wav2vec2.py",
          "tests/models/wav2vec2/test_processor_wav2vec2.py",
          "tests/models/wav2vec2/test_tokenization_wav2vec2.py",
      }
      self.assertEqual(test_files, wav2vec2_test_files)
  
imports: import os
  import re
  import tempfile
  import unittest
  from pathlib import Path
  import transformers
  from transformers.commands.add_new_model_like import ModelPatterns, _re_class_func, add_content_to_file, add_content_to_text, clean_frameworks_in_init, duplicate_doc_file, duplicate_module, filter_framework_files, find_base_model_checkpoint, get_model_files, get_module_from_file, parse_module_content, replace_model_patterns, retrieve_info_for_model, retrieve_model_classes, simplify_replacements
  from transformers.testing_utils import require_flax, require_tf, require_torch
label_desc: Use image node_id 15 for calling the TestAddNewModelLike obj's underlying member method code with example usage: obj.test_get_model_files_only_pt() without return types