this duplicated?

This commit is contained in:
yoni13 2025-01-14 19:22:28 +08:00
parent 0f03f77e8e
commit b5a4ed5160

View File

@ -362,7 +362,7 @@ class TestRknnSession:
[mock.call(f"Loaded RKNN model from {str(model_path).replace("model","rk3566")} with {tpe} threads.")] [mock.call(f"Loaded RKNN model from {str(model_path).replace("model","rk3566")} with {tpe} threads.")]
) )
def test_run(self, rknn_session: mock.Mock, mocker: MockerFixture) -> None: def test_run_rknn(self, rknn_session: mock.Mock, mocker: MockerFixture) -> None:
rknn_session.return_value.load.return_value = 123 rknn_session.return_value.load.return_value = 123
np_spy = mocker.spy(np, "ascontiguousarray") np_spy = mocker.spy(np, "ascontiguousarray")
mocker.patch("app.sessions.rknn.soc_name", "rk3566") mocker.patch("app.sessions.rknn.soc_name", "rk3566")