import numpy as np
import pytest

from objdetect_benchmark import get_norm, get_max_error, TypeNorm

def test_get_norm_l1():
    a = np.array([[0, 0], [1, 1]], float)
    b = np.array([[1, 1], [2, 2]], float)
    # |(1,1)-(0,0)|₁ + |(2,2)-(1,1)|₁ = (1+1) + (1+1) = 4
    assert get_norm(a, b, TypeNorm.l1) == pytest.approx(4)

def test_get_norm_l2():
    a = np.array([[0, 0], [0, 0]], float)
    b = np.array([[3, 4], [0, 0]], float)
    # flatten difference = [3,4,0,0], L₂ = sqrt(3²+4²) = 5
    assert get_norm(a, b, TypeNorm.l2) == pytest.approx(5)

def test_get_norm_linf():
    a = np.array([[0, 0], [1, 5]], float)
    b = np.array([[2, 1], [0, 0]], float)
    # per-point diffs: [(2,1),(1,5)] → flattened [2,1,1,5], max = 5
    assert get_norm(a, b, TypeNorm.l_inf) == pytest.approx(5)

def test_get_norm_iou_full_overlap():
    # identical quads → IoU = 1.0
    quad = np.array([[0,0], [1,0], [1,1], [0,1]], float)
    assert get_norm(quad, quad, TypeNorm.intersection_over_union) == pytest.approx(1.0)

def test_get_norm_iou_partial_overlap():
    # gold: unit square; detected: shifted right by 0.5
    gold = np.array([[0,0], [1,0], [1,1], [0,1]], float)
    det  = np.array([[0.5,0], [1.5,0], [1.5,1], [0.5,1]], float)
    # overlap area = 0.5×1 = 0.5; union = 1 + 1 - 0.5 = 1.5 → IoU = 1/3
    assert get_norm(gold, det, TypeNorm.intersection_over_union) == pytest.approx(1/3)

def test_get_max_error():
    assert get_max_error(5, TypeNorm.l2) == 5
    assert get_max_error(0.1, TypeNorm.intersection_over_union) == 1.0

import os
import pytest
from objdetect_benchmark import main

def test_gen_and_show(tmp_path, monkeypatch):
    monkeypatch.chdir(tmp_path)

    import sys
    sys.argv[:] = [
        "objdetect_benchmark.py",
        "--configuration", "generate",
        "-p", str(tmp_path/"out"),
        "--synthetic_object", "chessboard",
        "--cell_img_size", "50",
        "--board_x", "3", "--board_y", "3"
    ]

    main()

    output_files = list((tmp_path/"out").rglob("*.json"))
    assert output_files, "expected some JSON"