update template code
This commit is contained in:
53
tests/test_a4.py
Normal file
53
tests/test_a4.py
Normal file
@@ -0,0 +1,53 @@
|
||||
# -*- 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
|
||||
Reference in New Issue
Block a user