TEST: fix a test under recent numpys