Unit test are not just for people who write bugs (parent comment), a good start would be to test for the 9th row the problem was asking for. Since you can't test all possible values a representative sample would be good enough.
Just to be pedantic: It's not TDD if he already wrote the functional code ;)