import pytest

import os, glob

import cv2

import micasense.imageset as imageset

import micasense.capture as capture

import micasense.image as image

import micasense.imageutils as imageutils

from numpy import array

from numpy import float32

truth_warp_matrices = [array([[ 1.00523243e+00, -3.95214025e-03, -1.02620616e+01],

[-4.15804231e-06, -2.63551964e-06, 1.00000000e+00]])]

truth_image_sizes = [(2064, 1544), (2064, 1544), (2064, 1544), (2064, 1544), (2064, 1544), (160, 120)]

truth_lens_distortions = [array([-1.360334e-01, 2.374279e-01, 1.761687e-04, 2.373747e-04,

-4.892879e-01])]

truth_camera_matrices = [array([[2.24343510e+03, 0.00000000e+00, 1.01295942e+03],

[ 0. , 0. , 1. ]])]

expected_dimensions = (21.0, 12.0, 2035.0, 1467.0)

def test_image_properties(non_panel_altum_capture):

assert(image.cv2_camera_matrix() == pytest.approx(truth_camera_matrices[i]))

def test_warp_matrices(non_panel_altum_capture):

warp_matrices = non_panel_altum_capture.get_warp_matrices()

for index,warp_matrix in enumerate(warp_matrices):

assert(warp_matrix == pytest.approx(truth_warp_matrices[index],rel=1e-2))

def test_cropping(non_panel_altum_capture):

warp_matrices = non_panel_altum_capture.get_warp_matrices()

cropped_dimensions,_ = imageutils.find_crop_bounds(non_panel_altum_capture,warp_matrices)

assert(cropped_dimensions == pytest.approx(expected_dimensions,abs=1))