Added basic tests for transformRawStages, to
ensure all stage properties are retained.

Minor review feedback