Add unit tests for array source