test: Simplify generator test

The mutation tests are already checking that we sum the inputs at each iteration, so it's sufficient to check the first three results.