code: def base_model_context(base_model_dir):
      context = MockContext(
          model_name="mnist",
          model_dir=base_model_dir.as_posix(),
          model_file="model.py",
      )
      yield context
  
file_path: serve/ts/torch_handler/unit_tests/conftest.py
imports: import shutil
  import sys
  from pathlib import Path
  import pytest
  from .models.base_model import save_pt_file
  from .test_utils.mock_context import MockContext
label_desc: Use image node_id 2 for calling a global function with example usage: base_model_context(base_model_dir) without return types