Added test for the case of array input / output.