我想測驗以下函式，但仍然難以找到測驗 I/O 操作的最佳實踐。
def get_weight_file(path: Union[Path, str]) -> str:
"""Finds weights (.model) file in a directory
Parameters
----------
path: Union[Path, str]
Path where to find the weights file
Returns
-------
str
Filename of the weights (.model) file
"""
only_files = [
file for file in os.listdir(path) if os.path.isfile(os.path.join(path, file))
]
model_file = [file for file in only_files if file.endswith(".model")]
if len(model_file) == 0:
raise FileNotFoundError("No weights file found in current directory")
if len(model_file) > 1:
raise MultipleFilesError("Please provide a single weights file")
return model_file[0]
我試圖模擬 os.listdir。
@mock.patch("os.listdir", return_value=["test.model", "test.txt", "text.yaml"])
def test_get_weight_file(listdir):
assert get_weight_file(path="./") == "test.model"
這是錯誤：
if len(model_file) == 0:
> raise FileNotFoundError("No weights file found in current directory")
E FileNotFoundError: No weights file found in current directory
該函式似乎無法檢索“test.model”檔案。無論如何，它不起作用，我不知道為什么，我也懷疑我解決這個問題的方法是最佳實踐。誰能幫我解決這個問題？
uj5u.com熱心網友回復：
實際上，您應該 patch @mock.patch('os.listdir')，請參閱where-to-patch。
另外，我還打了補丁os.path.isfile()方法，因為我的真實檔案系統上沒有這些目錄檔案，我把它的回傳值模擬為True.
例如
get_weight_file.py：
import os
def get_weight_file(path):
only_files = [
file for file in os.listdir(path) if os.path.isfile(os.path.join(path, file))
]
print('os.listdir(path): ', os.listdir(path))
model_file = [file for file in only_files if file.endswith(".model")]
return model_file[0]
test_get_weight_file.py：
from unittest import TestCase, mock
import unittest
from get_weight_file import get_weight_file
class TestGetWeightFile(TestCase):
@mock.patch("os.path.isfile", return_value=True)
@mock.patch("os.listdir", return_value=["test.model", "test.txt", "text.yaml"])
def test_get_weight_file(self, mock_listdir, mock_isfile):
print('mock_isfile: ', mock_isfile.return_value)
print('mock_listdir: ', mock_listdir.return_value)
assert get_weight_file(path="./") == "test.model"
if __name__ == '__main__':
unittest.main()
測驗結果：
mock_isfile: True
mock_listdir: ['test.model', 'test.txt', 'text.yaml']
os.listdir(path): ['test.model', 'test.txt', 'text.yaml']
.
----------------------------------------------------------------------
Ran 1 test in 0.001s
OK
轉載請註明出處，本文鏈接：https://www.uj5u.com/qukuanlian/379991.html