Add test for 2-arg transform

The transform function is stream agnostic