Complete migrating enforce_distrubution tests to insta (#18185)

- Closes #15791
- Closes #15178 🥳

- Surpasses part of #16978

---------

Co-authored-by: Claude <[email protected]>
Co-authored-by: Andrew Lamb <[email protected]>

Author: 3 people