[TASK] Tests for the data transformation feature