當前位置: 首頁>>代碼示例 >>用法及示例精選 >>正文


Python tf.test.TestCase.create_tempdir用法及代碼示例


用法

create_tempdir(
    name=None, cleanup=None
)

參數

  • name 目錄的可選名稱。如果沒有給出,將生成並使用一個唯一的名稱。
  • cleanup 關於何時/是否在測試結束時刪除目錄(及其所有內容)的可選清理策略。如果沒有,則使用 self.tempfile_cleanup

返回

  • 一個 _TempDir 代表創建的目錄;有關用法,請參閱 _TempDir 類文檔。

創建一個特定於測試的臨時目錄。

注意:目錄及其內容將在創建前被遞歸清除。這確保不存在預先存在的狀態。

這將在磁盤上創建一個與此測試隔離的命名目錄,並將由測試正確清理。這避免了為測試目的創建臨時目錄的幾個陷阱,並且更容易設置目錄和驗證其內容。例如:

def test_foo(self):
  out_dir = self.create_tempdir()
  out_log = out_dir.create_file('output.log')
  expected_outputs = [
      os.path.join(out_dir, 'data-0.txt'),
      os.path.join(out_dir, 'data-1.txt'),
  ]
  code_under_test(out_dir)
  self.assertTrue(os.path.exists(expected_paths[0]))
  self.assertTrue(os.path.exists(expected_paths[1]))
  self.assertEqual('foo', out_log.read_text())

另請參閱:create_tempfile() 創建臨時文件。

相關用法


注:本文由純淨天空篩選整理自tensorflow.org大神的英文原創作品 tf.test.TestCase.create_tempdir。非經特殊聲明,原始代碼版權歸原作者所有,本譯文未經允許或授權,請勿轉載或複製。