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()
