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'], [])