add unit test
Merge pull request #103 from alberts-tid/task/add_unit_tests

add unit test