# -*- coding: utf-8 -*-

from datetime import datetime
from pathlib import Path
import pytest

from . import check_bad_imports
from mmp.a4 import anchor_grid, label_grid, dataset

current_assignment = pytest.mark.skipif(
    not (datetime(2025, 11, 6) <= datetime.now() <= datetime(2025, 11, 12, 23, 59, 59)),
    reason="This is not the current assignment.",
)


@current_assignment
def test_no_abs_import():
    paths = list(Path().glob("mmp/a4/*.py"))
    check_bad_imports(paths)


@current_assignment
def test_anchor_grid():
    grid = anchor_grid.get_anchor_grid(
        num_rows=4,
        num_cols=5,
        scale_factor=12.0,
        anchor_widths=[30.0, 80.0],
        aspect_ratios=[0.5, 1.0],
    )
    assert grid.ndim == 5
    assert grid.shape[-1] == 4


@current_assignment
def test_iou():
    label_grid.iou


@current_assignment
def test_label_grid():
    lg, *_ = label_grid.get_label_grid(
        anchor_grid=anchor_grid.get_anchor_grid(2, 2, 10, [10.0], [1.0]),
        gts=[],
        min_iou=0.5,
    )
    assert lg.dtype == bool


@current_assignment
def test_dataset():
    dataset.MMP_Dataset
    dataset.calculate_max_coverage