Added assertions for test cases
Fix new assertions, refs #16