add unit tests
5 files changed