Add extra asserts to tests
Merge pull request #8 from pratid/feature/unit_tests

Add extra asserts to tests