updated tests to assert results