* wip * Add tests * Update package * remove logs * nit * nit --------- Co-authored-by: huchenlei <huchenlei@proton.me>