Add test for array_key_(first|last|index)