def test_update_datasource(self, params): update_datasource = params["update"] datasource = params["add"] test_name = util.get_name(datasource["name"]) datasource_id = self.cruder.add_datasource( name=test_name, subid=datasource["subid"], creator=datasource["creator"], owner=datasource["owner"], current_process=datasource["current_process"], last_modifier=datasource["last_modifier"], database_name=datasource["database_name"], description=datasource['description'], folder_depth=datasource["folder_depth"], paramone=datasource["paramone"], password=datasource["password"], server_ip=datasource["server_ip"], server_port=datasource["server_port"], username=datasource["username"], workbench_url=datasource["workbench_url"]) self.cruder.update_datasource( datasource_id=datasource_id, server_port=update_datasource["server_port"]) datasource_db_instance = self.test_db_session.query(Datasource).get( datasource_id) # 断言 self.assertEqual(datasource_db_instance.server_port, update_datasource["server_port"]) # 删除新增的测试数据 self.test_db_session.delete(datasource_db_instance) self.test_db_session.commit()
def test_get_dataset(self, params): dataset = params["add"] test_name = util.get_name(dataset["name"]) dataset_id = self.cruder.add_dataset(test_name, datasource_id=dataset["datasource_id"], subid=dataset["subid"], creator=dataset["creator"], owner=dataset["owner"], current_process=dataset["current_process"], last_modifier=dataset["last_modifier"], data_file_format=dataset["data_file_format"], default_filter_string=dataset["default_filter_string"], description=dataset["description"], filter=dataset["filter"], patterns=dataset["patterns"], target_entity_class=dataset["target_entity_class"]) dataset_crud = self.cruder.get_dataset(dataset_id) # 断言 self.assertEqual(dataset_crud.name, test_name) self.assertEqual(dataset_crud.datasource_id, dataset["datasource_id"]) self.assertEqual(dataset_crud.subid, dataset["subid"]) self.assertEqual(dataset_crud.creator, dataset["creator"]) self.assertEqual(dataset_crud.owner, dataset["owner"]) self.assertEqual(dataset_crud.current_process, dataset["current_process"]) self.assertEqual(dataset_crud.last_modifier, dataset["last_modifier"]) self.assertEqual(dataset_crud.data_file_format, dataset["data_file_format"]) self.assertEqual(dataset_crud.default_filter_string, dataset["default_filter_string"]) self.assertEqual(dataset_crud.description, dataset["description"]) self.assertEqual(dataset_crud.filter, dataset["filter"]) self.assertEqual(dataset_crud.patterns, dataset["patterns"]) self.assertEqual(dataset_crud.target_entity_class, dataset["target_entity_class"]) # 删除新增的测试数据 self.test_db_session.delete(dataset_crud) self.test_db_session.commit()
def test_clean_algorithm(self, params): algorithm = params["add"] test_name = util.get_name(algorithm["name"]) algorithm_id = self.cruder.add_algorithm( name=test_name, display_name=algorithm["display_name"], description=algorithm["description"], entry_name=algorithm["entry_name"], algorithm_type=algorithm["algorithm_type"], hyperparameter_config=algorithm["hyperparameter_config"], available=algorithm["available"], train_input_pattern=algorithm["train_input_pattern"], train_output_pattern=algorithm["train_output_pattern"], model_input_pattern=algorithm["model_input_pattern"], model_output_pattern=algorithm["model_output_pattern"], runtime=algorithm["runtime"], learning=algorithm["learning"], package_id=algorithm["package_id"]) self.cruder.clean_algorithm(algorithm_id) # 断言 # 根据id查询 algorithm_db_instance = self.test_db_session.query(Algorithm).get( algorithm_id) self.assertEqual(algorithm_db_instance, None)
def test_update_algorithm(self, params): update_algorithm = params["update"] algorithm = params["add"] test_name = util.get_name(algorithm["name"]) algorithm_id = self.cruder.add_algorithm( name=test_name, display_name=algorithm["display_name"], description=algorithm["description"], entry_name=algorithm["entry_name"], algorithm_type=algorithm["algorithm_type"], hyperparameter_config=algorithm["hyperparameter_config"], available=algorithm["available"], train_input_pattern=algorithm["train_input_pattern"], train_output_pattern=algorithm["train_output_pattern"], model_input_pattern=algorithm["model_input_pattern"], model_output_pattern=algorithm["model_output_pattern"], runtime=algorithm["runtime"], learning=algorithm["learning"], package_id=algorithm["package_id"]) self.cruder.update_algorithm( algorithm_id=algorithm_id, description=update_algorithm["description"]) algorithm_db_instance = self.test_db_session.query(Algorithm).get( algorithm_id) # 断言 self.assertEqual(algorithm_db_instance.description, update_algorithm["description"]) # 删除新增的测试数据 self.test_db_session.delete(algorithm_db_instance) self.test_db_session.commit()
def test_delete_package(self, params): package = params["add"] test_name = util.get_name(package["name"]) package_id = self.cruder.add_package( name=test_name, package_source=package["package_source"], package_path=package["package_path"], description=package["description"]) self.cruder.delete_package(package_id) # 断言 package_db_instance = self.test_db_session.query(Package).get( package_id) self.assertEqual(package_db_instance, None)
def test_get_datasource(self, params): datasource = params["add"] test_name = util.get_name(datasource["name"]) datasource_id = self.cruder.add_datasource( name=test_name, subid=datasource["subid"], creator=datasource["creator"], owner=datasource["owner"], current_process=datasource["current_process"], last_modifier=datasource["last_modifier"], database_name=datasource["database_name"], description=datasource['description'], folder_depth=datasource["folder_depth"], paramone=datasource["paramone"], password=datasource["password"], server_ip=datasource["server_ip"], server_port=datasource["server_port"], username=datasource["username"], workbench_url=datasource["workbench_url"]) datasource_crud = self.cruder.get_datasource(datasource_id) # 断言 self.assertEqual(datasource_crud.subid, datasource["subid"]) self.assertEqual(datasource_crud.name, test_name) self.assertEqual(datasource_crud.creator, datasource["creator"]) self.assertEqual(datasource_crud.owner, datasource["owner"]) self.assertEqual(datasource_crud.current_process, datasource["current_process"]) self.assertEqual(datasource_crud.last_modifier, datasource["last_modifier"]) self.assertEqual(datasource_crud.database_name, datasource["database_name"]) self.assertEqual(datasource_crud.description, datasource["description"]) self.assertEqual(datasource_crud.folder_depth, datasource["folder_depth"]) self.assertEqual(datasource_crud.paramone, datasource["paramone"]) self.assertEqual(datasource_crud.password, datasource["password"]) self.assertEqual(datasource_crud.server_ip, datasource["server_ip"]) self.assertEqual(datasource_crud.server_port, datasource["server_port"]) self.assertEqual(datasource_crud.username, datasource["username"]) self.assertEqual(datasource_crud.workbench_url, datasource["workbench_url"]) # 删除新增的测试数据 self.test_db_session.delete(datasource_crud) self.test_db_session.commit()
def test_get_algorithm(self, params): algorithm = params["add"] test_name = util.get_name(algorithm["name"]) algorithm_id = self.cruder.add_algorithm( name=test_name, display_name=algorithm["display_name"], description=algorithm["description"], entry_name=algorithm["entry_name"], algorithm_type=algorithm["algorithm_type"], hyperparameter_config=algorithm["hyperparameter_config"], available=algorithm["available"], train_input_pattern=algorithm["train_input_pattern"], train_output_pattern=algorithm["train_output_pattern"], model_input_pattern=algorithm["model_input_pattern"], model_output_pattern=algorithm["model_output_pattern"], runtime=algorithm["runtime"], learning=algorithm["learning"], package_id=algorithm["package_id"]) algorithm_crud = self.cruder.get_algorithm(algorithm_id) # 断言 self.assertEqual(algorithm_crud.name, test_name) self.assertEqual(algorithm_crud.display_name, algorithm["display_name"]) self.assertEqual(algorithm_crud.description, algorithm["description"]) self.assertEqual(algorithm_crud.entry_name, algorithm["entry_name"]) self.assertEqual(algorithm_crud.algorithm_type, algorithm["algorithm_type"]) self.assertEqual(algorithm_crud.parameters, algorithm["hyperparameter_config"]) self.assertEqual(algorithm_crud.available, algorithm["available"]) self.assertEqual(algorithm_crud.alg_input_patterns, algorithm["train_input_pattern"]) self.assertEqual(algorithm_crud.alg_output_patterns, algorithm["train_output_pattern"]) self.assertEqual(algorithm_crud.model_input_patterns, algorithm["model_input_pattern"]) self.assertEqual(algorithm_crud.model_output_patterns, algorithm["model_output_pattern"]) self.assertEqual(algorithm_crud.runtime, algorithm["runtime"]) self.assertEqual(algorithm_crud.package_id, algorithm["package_id"]) # 删除新增的测试数据 self.test_db_session.delete(algorithm_crud) self.test_db_session.commit()
def test_get_all_package(self, params): package = params["add"] test_name = util.get_name(package["name"]) package_id = self.cruder.add_package( name=test_name, package_source=package["package_source"], package_path=package["package_path"], description=package["description"]) package_all = self.cruder.get_all_packages() packages_db_all = self.test_db_session.query(Package).all() # 断言 self.assertEqual(len(package_all), len(packages_db_all)) # 数据清理 package_db_instance = self.test_db_session.query(Package).get( package_id) self.test_db_session.delete(package_db_instance) self.test_db_session.commit()
def test_delete_dataset(self, params): dataset = params["add"] test_name = util.get_name(dataset["name"]) dataset_id = self.cruder.add_dataset(test_name, datasource_id=dataset["datasource_id"], subid=dataset["subid"], creator=dataset["creator"], owner=dataset["owner"], current_process=dataset["current_process"], last_modifier=dataset["last_modifier"], data_file_format=dataset["data_file_format"], default_filter_string=dataset["default_filter_string"], description=dataset["description"], filter=dataset["filter"], patterns=dataset["patterns"], target_entity_class=dataset["target_entity_class"]) self.cruder.delete_dataset(dataset_id) # 断言 # 根据id查询 dataset_db_instance = self.test_db_session.query(Dataset).get(dataset_id) self.assertEqual(dataset_db_instance, None)
def test_update_package(self, params): package = params["add"] update_package = params["update"] test_name = util.get_name(package["name"]) package_id = self.cruder.add_package( name=test_name, package_source=package["package_source"], package_path=package["package_path"], description=package["description"]) self.cruder.update_package(package_id=package_id, name=update_package["name"]) # 断言 package_db_instance = self.test_db_session.query(Package).get( package_id) self.assertEqual(package_db_instance.name, update_package["name"]) # 数据清理 self.test_db_session.delete(package_db_instance) self.test_db_session.commit()
def test_get_package(self, params): package = params["add"] test_name = util.get_name(package["name"]) package_id = self.cruder.add_package( name=test_name, package_source=package["package_source"], package_path=package["package_path"], description=package["description"]) package_instance = self.cruder.get_package(package_id) # 断言 self.assertEqual(package_instance.name, test_name) self.assertEqual(package_instance.package_source, package["package_source"]) self.assertEqual(package_instance.package_path, package["package_path"]) self.assertEqual(package_instance.description, package["description"]) # 数据清理 self.test_db_session.delete(package_instance) self.test_db_session.commit()
def test_get_all_dataset(self, params): dataset = params["add"] test_name = util.get_name(dataset["name"]) dataset_id = self.cruder.add_dataset(test_name, datasource_id=dataset["datasource_id"], subid=dataset["subid"], creator=dataset["creator"], owner=dataset["owner"], current_process=dataset["current_process"], last_modifier=dataset["last_modifier"], data_file_format=dataset["data_file_format"], default_filter_string=dataset["default_filter_string"], description=dataset["description"], filter=dataset["filter"], patterns=dataset["patterns"], target_entity_class=dataset["target_entity_class"]) dataset_cruds = self.cruder.get_all_dataset() dataset_db_all = self.test_db_session.query(Dataset).all() # # 断言 self.assertEqual(len(dataset_cruds), len(dataset_db_all)) # # 删除新增的测试数据 dataset_db_instance = self.test_db_session.query(Dataset).get(dataset_id) self.test_db_session.delete(dataset_db_instance) self.test_db_session.commit()