Add unit test for all methods

Signed-off-by: Ondrej Kupka <ondra.cap@gmail.com>