"""Tests para calculate_media_strategy."""

from calculate_media_strategy import calculate_media_strategy


def test_0_imagenes_text_only():
    assert calculate_media_strategy(0, 50) == "text_only"


def test_2_imagenes_100_lineas_extract():
    assert calculate_media_strategy(2, 100) == "extract"


def test_10_imagenes_20_lineas_full_page_vlm():
    assert calculate_media_strategy(10, 20) == "full_page_vlm"


def test_5_imagenes_100_lineas_full_page_vlm():
    assert calculate_media_strategy(5, 100) == "full_page_vlm"


def test_0_lineas_division_por_cero_evitada():
    assert calculate_media_strategy(3, 0) == "text_only"
