code: def test_remove_files_when_checkout(tmp_dir, dvc, scm):
      # add the file into a separate branch
      scm.checkout("branch", True)
      ret = main(["checkout", "--force"])
      assert ret == 0
      tmp_dir.dvc_gen(
          "file_in_a_branch", "random text", commit="add file"
      )
  
      # Checkout back in master
      scm.checkout("master")
      assert os.path.exists("file_in_a_branch")
  
      # Make sure `dvc checkout` removes the file
      # self.dvc.checkout()
      ret = main(["checkout", "--force"])
      assert ret == 0
      assert not os.path.exists("file_in_a_branch")
  
imports: import logging
  import os
  import shutil
  import stat
  import textwrap
  import pytest
  from dvc.cli import main
  from dvc.dvcfile import PROJECT_FILE, load_file
  from dvc.exceptions import CheckoutError, CheckoutErrorSuggestGit, NoOutputOrStageError
  from dvc.fs import system
  from dvc.stage.exceptions import StageFileDoesNotExistError
  from dvc.utils import relpath
  from dvc.utils.fs import remove
  from tests.utils import get_gitignore_content
label_desc: Use image node_id 4 for calling a global function with example usage: test_remove_files_when_checkout(tmp_dir, dvc, scm) without return types