Add tests for new IYielder functionality