from tests.helpers import FakeSlide

from pathaia.patches.visu import preview_from_queries

from pathaia.util.types import Coord

import numpy as np

def test_preview_from_queries():

    slide = FakeSlide()

    queries = []

    min_res = 512

    cell_size = 20

    thickness = 2

    color = (255, 255, 0)

    size = Coord(256)

    level = 1

    size_0 = size * slide.level_downsamples[level]

    slide_size = Coord(slide.dimensions)

    thickness = 2 * (thickness // 2) + 1

    res = slide_size / size_0 * (thickness + cell_size) + thickness

    thumb_w = max(min_res, res.x)

    thumb_h = max(min_res, res.y)

    thumb = np.array(slide.get_thumbnail((thumb_w, thumb_h)).convert("RGB"))

    assert (

        preview_from_queries(

            slide,

            queries,

            min_res=min_res,

            color=color,

            thickness=thickness,

            cell_size=cell_size,

            size_0=size_0,

        )

        == thumb

    ).all()