kunit.py: functional changes for testing

Have the output be returned instead of printed directly so we can unit
test individual components.

Change-Id: Id3fa29bc26b5ecb987e68be278ffde3ed0730105
Signed-off-by: Felix Guo <felixguo@google.com>
2 files changed