Add tests for pick and omit.