Add tests for transform