Test array expression NumPy math