Refactor distribution tests to not be one huge file

This will make it easier to add new tests in the future.