Updates tests from previous commit to assert statements.