TST: make test work on recent numpy