Cleanup test_maps.py.

Changes maps.Distribution -> Distribution.

Clean up formatting.