Add tests for arg_find_runs and 'flat' case.