Add regression suit tests for subquery partitioning cases