Add tests for `pick` and `omit` operators