Improve array tests