зеркало из
https://github.com/ssciwr/AMMICO.git
synced 2025-10-30 05:26:05 +02:00
27732c5fab
* add py test * update dependencies and markers * Update ci.yml * fix linting * trying with editable install for cov * fix code smells * fix code smells Co-authored-by: xianghe ma <maxianghe@outlook.com>
64 строки
2.1 KiB
Python
64 строки
2.1 KiB
Python
import misinformation.cropposts as crpo
|
|
import numpy as np
|
|
from PIL import Image
|
|
|
|
TEST_IMAGE_1 = "./test/data/pic1.png"
|
|
TEST_IMAGE_2 = "./test/data/pic2.png"
|
|
|
|
|
|
def test_matching_points():
|
|
ref_view = np.array(Image.open(TEST_IMAGE_2))
|
|
view = np.array(Image.open(TEST_IMAGE_1))
|
|
filtered_matches, kp1, kp2 = crpo.matching_points(ref_view, view)
|
|
assert len(filtered_matches) > 0
|
|
|
|
|
|
def test_kp_from_matches():
|
|
ref_view = np.array(Image.open(TEST_IMAGE_2))
|
|
view = np.array(Image.open(TEST_IMAGE_1))
|
|
filtered_matches, kp1, kp2 = crpo.matching_points(ref_view, view)
|
|
kp1, kp2 = crpo.kp_from_matches(filtered_matches, kp1, kp2)
|
|
|
|
assert kp1.shape[0] == len(filtered_matches)
|
|
assert kp2.shape[0] == len(filtered_matches)
|
|
assert kp1.shape[1] == 2
|
|
assert kp2.shape[1] == 2
|
|
|
|
|
|
def test_compute_crop_corner():
|
|
ref_view = np.array(Image.open(TEST_IMAGE_2))
|
|
view = np.array(Image.open(TEST_IMAGE_1))
|
|
filtered_matches, kp1, kp2 = crpo.matching_points(ref_view, view)
|
|
corner = crpo.compute_crop_corner(filtered_matches, kp1, kp2)
|
|
print(view.shape)
|
|
print(corner)
|
|
assert corner is not None
|
|
v, h = corner
|
|
assert 0 <= v < view.shape[0]
|
|
assert 0 <= h < view.shape[0]
|
|
|
|
|
|
def test_crop_posts_image():
|
|
ref_view = np.array(Image.open(TEST_IMAGE_2))
|
|
view = np.array(Image.open(TEST_IMAGE_1))
|
|
rte = crpo.crop_posts_image(ref_view, view)
|
|
assert rte is not None
|
|
crop_view, match_num = rte
|
|
assert match_num > 0
|
|
assert crop_view.shape[0] * crop_view.shape[1] <= view.shape[0] * view.shape[1]
|
|
|
|
|
|
def test_crop_posts_from_refs():
|
|
ref_view = np.array(Image.open(TEST_IMAGE_2))
|
|
view = np.array(Image.open(TEST_IMAGE_1))
|
|
ref_views = [ref_view]
|
|
crop_view = crpo.crop_posts_from_refs(ref_views, view)
|
|
assert crop_view.shape[0] * crop_view.shape[1] <= view.shape[0] * view.shape[1]
|
|
|
|
|
|
def test_get_file_list():
|
|
ref_list = []
|
|
ref_dir = "./test/data"
|
|
ref_list = crpo.get_file_list(ref_dir, ref_list, ext="png")
|
|
assert len(ref_list) > 0
|