Better error reporting in test_math.py