Add test for partial selection