Add simple test for transforms layer