Adding unit test to array partitioner logic
Fixing logic issue on array partitioner