"""Unit tests for constants module."""

from analytics.consts.analytics import EntityTypes


def test_getting_entity_by_name():
    """Try getting the entity by its name."""
    assert EntityTypes.get_entity_from_name('label') == 'L'
    assert EntityTypes.get_entity_from_name('subaccount') == 'S'
    assert EntityTypes.get_entity_from_name('album') == 'A'