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

Add extra asserts to tests