Missing unit tests

here are some missing unit tests. feel free to reject PR with comments what should I do better :)
Code style!