added some unit tests for the layers fix when a ParallelDo is the last node of the graph