Add tests for math.py