[#5] Increase output.py test coverage