In the task description:

"Implement a function, which takes a non-negative integer"

So non-negative is assumed already. In production code this obviously would want some additional robustness, but based on the description of the task the solution fits.

The tests pass, and those are the final say anyway.