I think it is neater to split each invocation of the function under test to its own test function. Thus, when a test fails, it will describe under what conditions automatically. This is only a suggestion.