fix: reformat tests
fix: add unit tests