Add tests around output attribute