Add more tests: gather, reduce, transform reduce