update unit tests for transform_2D.py