Tests made more generic, Tree#draw given context parameter