Add numpy guard to tests