1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113 114 115 116 117 118 119 120 121 122 123 124 125 126 127 128 129 130 131 132 133 134 135 136 137 138 139 140 141 142 143 144 145 146 147 148 149 150 151 152 153 154 155 156 157 158 159 160 161 162 163 164 165 166
|
import cv2
import unittest
from dyda_utils import tools
from dyda_utils import tinycv
from dyda_utils import lab_tools
from dyda_utils import dict_comparator
from dyda.components.image_processor import CropRoiImageProcessor
# pull test data from gitlab
print('[Test_CropRoiImageProcessor] INFO: Pull 34 KB files from gitlab. ')
input_url = 'https://gitlab.com/DT42/galaxy42/dt42-dyda/uploads/'\
'05e15e0a9133b30f5bfbe6c02d6847a4/input_img.png.0'
input_data = lab_tools.pull_img_from_gitlab(input_url)
input_url = 'https://gitlab.com/DT42/galaxy42/dt42-dyda/uploads/'\
'63ee0fac22eb43882dcefa3ec154df54/frame_781.jpg.0'
input_data_calibrate = lab_tools.pull_img_from_gitlab(input_url)
config_url = 'https://gitlab.com/DT42/galaxy42/dt42-dyda/uploads/'\
'78a5e35b2802ddbee8614aefbceb2fdb/dyda.config.CropRoiImageProcessor'
dyda_config_CropRoiImageProcessor = lab_tools.pull_json_from_gitlab(
config_url)
metadata_url = 'https://gitlab.com/DT42/galaxy42/dt42-dyda/uploads/'\
'ddc0b4970202e99c3a1248c5af66480e/TestResizeImageProcessor_metadata.json'
metadata_CropRoiImageProcessor = lab_tools.pull_json_from_gitlab(metadata_url)
output_url = 'https://gitlab.com/DT42/galaxy42/dt42-dyda/uploads/'\
'c8c952e97929e1e3f47c15740f2b7dc4/TestCropRoiImageProcessor_simple.bmp.0'
output_CropRoiImageProcessor = lab_tools.pull_img_from_gitlab(output_url)
results_url = 'https://gitlab.com/DT42/galaxy42/dt42-dyda/uploads/'\
'473726f1c287475360e7a177dbb2b4aa/TestCropRoiImageProcessor_simple.json'
results_CropRoiImageProcessor = lab_tools.pull_json_from_gitlab(results_url)
output_url = 'https://gitlab.com/DT42/galaxy42/dt42-dyda/uploads/'\
'a9bf448d4c9831559053bde0dea0c595/TestCropRoiImageProcessor_metadata.bmp.0'
output_CropRoiImageProcessor_m = lab_tools.pull_img_from_gitlab(output_url)
results_url = 'https://gitlab.com/DT42/galaxy42/dt42-dyda/uploads/'\
'eafb01647855d671bfa3e1c19596f9d7/TestCropRoiImageProcessor_metadata.json'
results_CropRoiImageProcessor_m = lab_tools.pull_json_from_gitlab(results_url)
output_url = 'https://gitlab.com/DT42/galaxy42/dt42-dyda/uploads/'\
'42a4b5bc1dc5f0ec8aad0ac58abdb699/TestResizeImageProcessor_simple.bmp.0'
output_ResizeImageProcessor = lab_tools.pull_img_from_gitlab(output_url)
class TestCropRoiImageProcessor_metadata(unittest.TestCase):
""" Test if ResizeImageProcessor is applied before. """
def test_main_process(self):
""" Main process of unit test. """
# initialization
comp = CropRoiImageProcessor(
dyda_config_path=dyda_config_CropRoiImageProcessor)
# run component
comp.reset()
comp.metadata = metadata_CropRoiImageProcessor
comp.input_data = output_ResizeImageProcessor
comp.run()
# compare output_data with reference
ref_data = output_CropRoiImageProcessor_m
tar_data = comp.output_data
img_diff = lab_tools.img_comparator(tar_data, ref_data)
self.assertEqual(img_diff, 0.0)
# compare results with reference
ref_data = results_CropRoiImageProcessor_m
tar_data = comp.results
report = dict_comparator.get_diff(ref_data, tar_data)
self.assertEqual(report['extra_field'], [])
self.assertEqual(report['missing_field'], [])
self.assertEqual(report['mismatch_val'], [])
class TestCropRoiImageProcessor_simple(unittest.TestCase):
""" Test simple case. """
def test_main_process(self):
""" Main process of unit test. """
# initialization
comp = CropRoiImageProcessor(
dyda_config_path=dyda_config_CropRoiImageProcessor)
# run component
comp.reset()
comp.input_data = input_data
comp.run()
# compare output_data with reference
ref_data = output_CropRoiImageProcessor
tar_data = comp.output_data
img_diff = lab_tools.img_comparator(tar_data, ref_data)
self.assertEqual(img_diff, 0.0)
# compare results with reference
ref_data = results_CropRoiImageProcessor
tar_data = comp.results
report = dict_comparator.get_diff(ref_data, tar_data)
self.assertEqual(report['extra_field'], [])
self.assertEqual(report['missing_field'], [])
self.assertEqual(report['mismatch_val'], [])
class TestCropRoiImageProcessor_double(unittest.TestCase):
""" Double test simple case. """
def test_main_process(self):
""" Main process of unit test. """
# initialization
comp = CropRoiImageProcessor(
dyda_config_path=dyda_config_CropRoiImageProcessor)
# run component
comp.reset()
comp.input_data = input_data
comp.run()
comp.input_data = input_data
comp.run()
# compare output_data with reference
ref_data = output_CropRoiImageProcessor
tar_data = comp.output_data
img_diff = lab_tools.img_comparator(tar_data, ref_data)
self.assertEqual(img_diff, 0.0)
# compare results with reference
ref_data = results_CropRoiImageProcessor
tar_data = comp.results
report = dict_comparator.get_diff(ref_data, tar_data)
self.assertEqual(report['extra_field'], [])
self.assertEqual(report['missing_field'], [])
self.assertEqual(report['mismatch_val'], [])
class TestCropRoiImageProcessor_list(unittest.TestCase):
""" Test list of input. """
def test_main_process(self):
""" Main process of unit test. """
# initialization
comp = CropRoiImageProcessor(
dyda_config_path=dyda_config_CropRoiImageProcessor)
# run component
comp.reset()
comp.input_data = [input_data]
comp.run()
# compare output_data with reference
ref_data = output_CropRoiImageProcessor
tar_data = comp.output_data[0]
img_diff = lab_tools.img_comparator(tar_data, ref_data)
self.assertEqual(img_diff, 0.0)
# compare results with reference
ref_data = results_CropRoiImageProcessor
tar_data = comp.results[0]
report = dict_comparator.get_diff(ref_data, tar_data)
self.assertEqual(report['extra_field'], [])
self.assertEqual(report['missing_field'], [])
self.assertEqual(report['mismatch_val'], [])
if __name__ == '__main__':
unittest.main()
|