Yeah, at least the comment should reflect both cases, but it's [more than just a comment](https://docs.python.org/2/library/doctest.html).