Enhancing GroupingAndPartitioningTest.java