Add tests for math operations