Add unit tests for Output class