add tests

- [x] add tests