improved NumPy tests