add more asserts to tests