test(annotation): cover module distribution

[UkiStefanovic]

This pr adds new tests to cover the module annotation/distribution/main.py

All unit tests for the function in this module can be found in /tests/test_distribution.py