add test,founds bugs, fix it tomorrow

2025-01-14 01:08:44 +08:00
parent 6c4e6cb96f
commit 4b0f93cf6a
3 changed files with 84 additions and 2 deletions
@@ -136,6 +136,12 @@ def ann_session() -> Iterator[mock.Mock]:
        yield mocked


+@pytest.fixture(scope="function")
+def rknn_session() -> Iterator[mock.Mock]:
+    with mock.patch("app.sessions.rknn.rknnPoolExecutor") as mocked:
+        yield mocked
+
+
@pytest.fixture(scope="function")
 def rmtree() -> Iterator[mock.Mock]:
    with mock.patch("app.models.base.rmtree", autospec=True) as mocked: