Added partial unit tests

Here are the tests written on testfest
See comment above