REF: Parametrize test
Let me know if I'm mistaken here, but I think we would prefer parametrization over for loops inside tests (e.g., so `pytest` flags which case is failing), and this should be equivalent.