def test_main_process(self):
""" Main process of unit test. """
# test case with plotting arrow
dyda_config_LocationBubblePlotter = {
'LocationBubblePlotter': {
"belongs_to": "virtual_id",
"plot_arrow": True
}
}
output_url = 'https://gitlab.com/DT42/galaxy42/dt42-dyda/'\
'uploads/ee7e62a737ab364cf96bc29dfaf55425/bp_arrow.bmp'
# initialization
comp = LocationBubblePlotter(
dyda_config_path=dyda_config_LocationBubblePlotter)
comp.reset()
comp.input_data = test_df
comp.run()
# compare output_data with reference
ref_data = lab_tools.pull_img_from_gitlab(output_url)
tar_data = comp.output_data
img_diff = lab_tools.img_comparator(ref_data, tar_data)
self.assertEqual(img_diff, 0.0)
# test case without plotting arrow
dyda_config_LocationBubblePlotter = {
'LocationBubblePlotter': {
"belongs_to": "virtual_id",
"plot_arrow": False
}
}
output_url = 'https://gitlab.com/DT42/galaxy42/dt42-dyda/'\
'uploads/4b592ecb09944b7d9412db84a233f166/bp.bmp'
# initialization
comp = LocationBubblePlotter(
dyda_config_path=dyda_config_LocationBubblePlotter)
comp.reset()
comp.input_data = test_df
comp.run()
# compare output_data with reference
ref_data = lab_tools.pull_img_from_gitlab(output_url)
tar_data = comp.output_data
img_diff = lab_tools.img_comparator(ref_data, tar_data)
self.assertEqual(img_diff, 0.0)
def test_main_process(self):
""" Main process of unit test. """
dyda_config_DataFrameHistPlotter = {
'DataFrameHistPlotter': {
'hist_feature': ['longitude', 'latitude']
}
}
output_url = 'https://gitlab.com/DT42/galaxy42/dt42-dyda/'\
'uploads/23e072aff6360991f5c004b475aaf7f2/hist.bmp'
# initialization
comp = DataFrameHistPlotter(
dyda_config_path=dyda_config_DataFrameHistPlotter)
# run component
comp.reset()
comp.input_data = test_df
comp.run()
# compare output_data with reference
ref_data = lab_tools.pull_img_from_gitlab(output_url)
tar_data = comp.output_data
img_diff = lab_tools.img_comparator(ref_data, tar_data)
self.assertEqual(img_diff, 0.0)
# test the case with using all parameters
dyda_config_DataFrameHistPlotter = {
'DataFrameHistPlotter': {
'hist_feature': ['longitude', 'latitude'],
'range': [(-122.5, -121.5), (47.5, 47.7)],
'bins': 20,
'belongs_to': 'virtual_id'
}
}
output_url = 'https://gitlab.com/DT42/galaxy42/dt42-dyda/uploads/'\
'c7667cbeae10c6cc41defcae2c6bafe1/hist_complicated.bmp'
# initialization
comp = DataFrameHistPlotter(
dyda_config_path=dyda_config_DataFrameHistPlotter)
# run component
comp.reset()
comp.input_data = test_df
comp.run()
# compare output_data with reference
ref_data = lab_tools.pull_img_from_gitlab(output_url)
tar_data = comp.output_data
img_diff = lab_tools.img_comparator(ref_data, tar_data)
self.assertEqual(img_diff, 0.0)
def test_main_process(self):
""" Main process of unit test. """
# initialization
comp = ResizeImageProcessor(
dyda_config_path=dyda_config_ResizeImageProcessor)
# run component
comp.reset()
comp.input_data = [input_data]
comp.run()
# compare output_data with reference
ref_data = output_ResizeImageProcessor
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_ResizeImageProcessor
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'], [])
def test_main_process(self):
""" Main process of unit test. """
# Test simple case
dyda_config = {"BGR2RGBImageProcessor": {}}
# initialization
comp = BGR2RGBImageProcessor(dyda_config_path=dyda_config)
# run component
comp.reset()
comp.input_data = input_data.copy()
comp.run()
# compare output_data with reference
ref_data = cv2.cvtColor(input_data.copy(), cv2.COLOR_BGR2RGB)
tar_data = comp.output_data
img_diff = lab_tools.img_comparator(tar_data, ref_data)
self.assertEqual(img_diff, 0.0)
# Test list input case
dyda_config = {"BGR2RGBImageProcessor": {}}
# initialization
comp = BGR2RGBImageProcessor(dyda_config_path=dyda_config)
# run component
comp.reset()
comp.input_data = [input_data.copy()]
comp.run()
# compare output_data with reference
ref_datas = [cv2.cvtColor(input_data.copy(), cv2.COLOR_BGR2RGB)]
tar_datas = comp.output_data
for ref_data, tar_data in zip(ref_datas, tar_datas):
img_diff = lab_tools.img_comparator(tar_data, ref_data)
self.assertEqual(img_diff, 0.0)
def test_main_process(self):
""" Main process of unit test. """
# initialization
comp = BGR2GrayImageProcessor(
dyda_config_path=dyda_config_BGR2GrayImageProcessor)
# run component
comp.reset()
comp.input_data.append(input_data)
comp.run()
# compare output_data with reference
ref_data = output_BGR2GrayImageProcessor
tar_data = comp.output_data[0]
img_diff = lab_tools.img_comparator(tar_data, ref_data)
self.assertEqual(img_diff, 0.0)
def test_main_process(self):
""" Main process of unit test. """
# initialization
comp = MergeImageProcessor(
dyda_config_path=dyda_config_MergeImageProcessor)
# run component
comp.reset()
comp.input_data = [
input_data_0, input_data_1, input_data_2, input_data_3
]
comp.run()
# compare output_data with reference
ref_data = output_MergeImageProcessor
tar_data = comp.output_data
img_diff = lab_tools.img_comparator(tar_data, ref_data)
self.assertEqual(img_diff, 0.0)
def test_main_process(self):
""" Main process of unit test. """
# initialization
comp = ExtractNonBlackImageProcessor()
# run component
comp.reset()
comp.input_data = input_data
comp.run()
# compare output_data with reference
tar_data = comp.output_data
img_diff = lab_tools.img_comparator(tar_data, ref_output)
self.assertEqual(img_diff, 0.0)
# compare results with reference
tar_data = comp.results
report = dict_comparator.get_diff(ref_results, tar_data)
self.assertEqual(report['extra_field'], [])
self.assertEqual(report['missing_field'], [])
self.assertEqual(report['mismatch_val'], [])
