Add test for selecting data