Do Unit Testing data files support testing of all DataStage stage types?

Yes. When a MettleCI unit test is executed it replaces all the source stages with the relevant data file referenced in the MettleCI unit test specification.  It also compares the output of each target stage against the expected output file referenced in the specification. This is true for all stage types, including custom stages that you might code yourself using the C++ API.