Add more array api tests