- Change the code to keep current return values / behavior
- Adapt the unit test to fit the new code
- Write a less complex patch for just adding the option
"--keep-partition-images" without migration to argparse
It looks like the exit 2 is deeply nested into the argparse code, so I guess the test should be updated. The test could be made more resilient by checking that the output wasn't 0 instead of was 1.