def send_and_compare_result(app_client, url, body_data, expect_file=None, method='post', full_url=False): """Send and compare result.""" res = get_request_result(app_client, url, body_data, method=method, full_url=full_url) delete_random_items(res) if expect_file: real_path = os.path.join(DEBUGGER_EXPECTED_RESULTS, 'restful_results', expect_file) compare_result_with_file(res, real_path)
def test_get_timeline_summary(self, device_target, filename): """Test the function of get timeline summary data for UI display.""" file_path = os.path.join(self.profiler, filename.format(self.device_id)) result = self._analyser.get_timeline_summary(device_target) compare_result_with_file(result, file_path)
def test_search_nodes_by_type(self, node_type, condition, result_file): """Test search nodes by type.""" result = self.graph_handler.search_nodes({ 'node_category': node_type, 'condition': condition, 'graph_name': 'kernel_graph_0' }) file_path = os.path.join(self.graph_results_dir, result_file) compare_result_with_file(result, file_path)
def test_search_nodes_by_type(self, node_type, condition, result_file): """Test search nodes by type.""" search_nodes = self.graph_handler.get_searched_node_list( { 'node_category': node_type, 'condition': condition }, 'kernel_graph_0') file_path = os.path.join(self.graph_results_dir, result_file) result = {'node_names': [node.name for node in search_nodes]} compare_result_with_file(result, file_path)
def test_get_display_timeline(self): """Test the function of get timeline detail data for UI display.""" gpu_file_path = os.path.join( self.profiler, self.gpu_display_filename.format(self.device_id)) ascend_file_path = os.path.join( self.profiler, self.ascend_display_filename.format(self.device_id)) result = self._analyser.get_display_timeline("gpu", 0) compare_result_with_file(result, gpu_file_path) result = self._analyser.get_display_timeline("ascend", 0) compare_result_with_file(result, ascend_file_path)
def compare_debugger_result_with_file(res, expect_file, save=False): """ Compare debugger result with file. Args: res (dict): The debugger result in dict type. expect_file: The expected file name. """ real_path = os.path.join(DEBUGGER_EXPECTED_RESULTS, expect_file) if save: with open(real_path, 'w') as file_handler: json.dump(res, file_handler) else: compare_result_with_file(res, real_path)
def test_load_device_info_ascend(self): """Test load_device_info of ascend chip for offline-debugger.""" res = self.data_loader_ascend.load_device_info() expected_result = os.path.join(self.expected_results_dir, 'load_device_info_ascend.json') compare_result_with_file(res, expected_result)
def test_get(self, filter_condition, result_file): """Test get.""" reply = self.handler.get(filter_condition) file_path = os.path.join(self.results_dir, result_file) compare_result_with_file(reply, file_path)
def test_get_without_filter(self, filter_condition, result_file): """Test get with filer_condition is False.""" file_path = os.path.join(self.results_dir, result_file) reply = self.handler.get(filter_condition) watch_points = reply.get('watch_points') compare_result_with_file(watch_points, file_path)
def test_search_nodes(self, pattern, result_file): """Test search nodes.""" result = self.graph_handler.search_nodes({'name': pattern}) file_path = os.path.join(self.graph_results_dir, result_file) compare_result_with_file(result, file_path)
def test_get_tensor_history(self, node_name, result_file): """Test get tensor history.""" result = self.graph_handler.get_tensor_history(node_name) file_path = os.path.join(self.graph_results_dir, result_file) compare_result_with_file(result, file_path)
def test_get(self, filter_condition, result_file): """Test get.""" result = self.graph_handler.get(filter_condition) file_path = os.path.join(self.graph_results_dir, result_file) compare_debugger_result_with_file(result, file_path, True) compare_result_with_file(result, file_path)