from beatos_core.assets import ASSET_ROLES, AUDIO_ROLES


def test_asset_roles_set():
    assert ASSET_ROLES == frozenset({
        "cover",
        "stems",
        "loop",
        "audio_tagged_mp3",
        "audio_untagged_mp3",
        "audio_tagged_wav",
        "audio_untagged_wav",
        "promo_video_vertical",
        "promo_video_landscape",
        "promo_video_square",
    })


def test_audio_roles_subset():
    assert AUDIO_ROLES == frozenset({
        "audio_tagged_mp3",
        "audio_untagged_mp3",
        "audio_tagged_wav",
        "audio_untagged_wav",
        "loop",
    })
    assert AUDIO_ROLES.issubset(ASSET_ROLES)
