update tests for new math features