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>