added asserts to tests