Add test case for new splits implementation