But I'm testing total_rows on every line? Or do you mean I should be testing cases where total_rows actually changes?