Skip to content

Commit 78e4d6e

Browse files
wendyJingwendyJing
committed
commit testcase
1 parent 8a24b73 commit 78e4d6e

19 files changed

+1834
-1
lines changed

‎requirements.txt

+4-1
Original file line numberDiff line numberDiff line change
@@ -10,4 +10,7 @@ scipy>=1.2.1
1010
six>=1.12.0
1111
Flask>=1.0.2
1212
sqlalchemy>=1.2
13-
torch>=1.0
13+
torch>=1.0
14+
pyyaml>=5.1
15+
ddt>=1.2.1
16+
parameterized>=0.7.0

‎test/cases/__init__.py

Whitespace-only changes.

‎test/cases/test_algorithm_crud.py

+265
Large diffs are not rendered by default.

‎test/cases/test_dataset_crud.py

+166
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,166 @@
1+
2+
import unittest
3+
import ddt
4+
from test.util.handle_yaml import HandleYaml
5+
from test.util import util
6+
7+
from dwf.common.config import test_config
8+
from dwf.ormmodels import build_test_session, Dataset
9+
from dwf.dataset.metadata.dataset_crud import DatasetCRUD
10+
11+
# 从yaml文件中取出该接口的参数
12+
params = HandleYaml("dataset_data.yaml").get_data()
13+
14+
15+
@ddt.ddt
16+
class TestDataset(unittest.TestCase):
17+
18+
@classmethod
19+
def setUpClass(cls):
20+
cls.test_db_session = build_test_session(test_config)
21+
cls.cruder = DatasetCRUD(db_session=cls.test_db_session)
22+
23+
@classmethod
24+
def tearDownClass(cls):
25+
cls.test_db_session.close_all()
26+
27+
@ddt.data(params)
28+
def test_add_dataset(self, params):
29+
dataset = params["add"]
30+
test_name = util.get_name(dataset["name"])
31+
dataset_id = self.cruder.add_dataset(test_name, datasource_id=dataset["datasource_id"], subid=dataset["subid"],
32+
creator=dataset["creator"], owner=dataset["owner"],
33+
current_process=dataset["current_process"],
34+
last_modifier=dataset["last_modifier"],
35+
data_file_format=dataset["data_file_format"],
36+
default_filter_string=dataset["default_filter_string"],
37+
description=dataset["description"],
38+
filter=dataset["filter"], patterns=dataset["patterns"],
39+
target_entity_class=dataset["target_entity_class"])
40+
# 断言
41+
dataset_db_instance = self.test_db_session.query(Dataset).get(dataset_id)
42+
43+
self.assertEqual(dataset_db_instance.name, test_name)
44+
self.assertEqual(dataset_db_instance.datasource_id, dataset["datasource_id"])
45+
self.assertEqual(dataset_db_instance.subid, dataset["subid"])
46+
self.assertEqual(dataset_db_instance.creator, dataset["creator"])
47+
self.assertEqual(dataset_db_instance.owner, dataset["owner"])
48+
self.assertEqual(dataset_db_instance.current_process, dataset["current_process"])
49+
self.assertEqual(dataset_db_instance.last_modifier, dataset["last_modifier"])
50+
self.assertEqual(dataset_db_instance.data_file_format, dataset["data_file_format"])
51+
self.assertEqual(dataset_db_instance.default_filter_string, dataset["default_filter_string"])
52+
self.assertEqual(dataset_db_instance.description, dataset["description"])
53+
self.assertEqual(dataset_db_instance.filter, dataset["filter"])
54+
self.assertEqual(dataset_db_instance.patterns, dataset["patterns"])
55+
self.assertEqual(dataset_db_instance.target_entity_class, dataset["target_entity_class"])
56+
57+
self.test_db_session.delete(dataset_db_instance)
58+
self.test_db_session.commit()
59+
60+
@ddt.data(params)
61+
def test_update_dataset(self, params):
62+
update_dataset = params["update"]
63+
dataset = params["add"]
64+
test_name = util.get_name(dataset["name"])
65+
dataset_id = self.cruder.add_dataset(test_name, datasource_id=dataset["datasource_id"], subid=dataset["subid"],
66+
creator=dataset["creator"], owner=dataset["owner"],
67+
current_process=dataset["current_process"],
68+
last_modifier=dataset["last_modifier"],
69+
data_file_format=dataset["data_file_format"],
70+
default_filter_string=dataset["default_filter_string"],
71+
description=dataset["description"],
72+
filter=dataset["filter"], patterns=dataset["patterns"],
73+
target_entity_class=dataset["target_entity_class"])
74+
75+
self.cruder.update_dataset(dataset_id=dataset_id, description=update_dataset["description"])
76+
77+
dataset_db_instance = self.test_db_session.query(Dataset).get(dataset_id)
78+
79+
# 断言
80+
self.assertEqual(dataset_db_instance.description, update_dataset["description"])
81+
# 删除新增的测试数据
82+
self.test_db_session.delete(dataset_db_instance)
83+
self.test_db_session.commit()
84+
85+
@ddt.data(params)
86+
def test_get_dataset(self, params):
87+
dataset = params["add"]
88+
test_name = util.get_name(dataset["name"])
89+
dataset_id = self.cruder.add_dataset(test_name, datasource_id=dataset["datasource_id"], subid=dataset["subid"],
90+
creator=dataset["creator"], owner=dataset["owner"],
91+
current_process=dataset["current_process"],
92+
last_modifier=dataset["last_modifier"],
93+
data_file_format=dataset["data_file_format"],
94+
default_filter_string=dataset["default_filter_string"],
95+
description=dataset["description"],
96+
filter=dataset["filter"], patterns=dataset["patterns"],
97+
target_entity_class=dataset["target_entity_class"])
98+
99+
dataset_crud = self.cruder.get_dataset(dataset_id)
100+
101+
# 断言
102+
self.assertEqual(dataset_crud.name, test_name)
103+
self.assertEqual(dataset_crud.datasource_id, dataset["datasource_id"])
104+
self.assertEqual(dataset_crud.subid, dataset["subid"])
105+
self.assertEqual(dataset_crud.creator, dataset["creator"])
106+
self.assertEqual(dataset_crud.owner, dataset["owner"])
107+
self.assertEqual(dataset_crud.current_process, dataset["current_process"])
108+
self.assertEqual(dataset_crud.last_modifier, dataset["last_modifier"])
109+
self.assertEqual(dataset_crud.data_file_format, dataset["data_file_format"])
110+
self.assertEqual(dataset_crud.default_filter_string, dataset["default_filter_string"])
111+
self.assertEqual(dataset_crud.description, dataset["description"])
112+
self.assertEqual(dataset_crud.filter, dataset["filter"])
113+
self.assertEqual(dataset_crud.patterns, dataset["patterns"])
114+
self.assertEqual(dataset_crud.target_entity_class, dataset["target_entity_class"])
115+
# 删除新增的测试数据
116+
self.test_db_session.delete(dataset_crud)
117+
self.test_db_session.commit()
118+
119+
@ddt.data(params)
120+
def test_get_all_dataset(self, params):
121+
dataset = params["add"]
122+
test_name = util.get_name(dataset["name"])
123+
dataset_id = self.cruder.add_dataset(test_name, datasource_id=dataset["datasource_id"], subid=dataset["subid"],
124+
creator=dataset["creator"], owner=dataset["owner"],
125+
current_process=dataset["current_process"],
126+
last_modifier=dataset["last_modifier"],
127+
data_file_format=dataset["data_file_format"],
128+
default_filter_string=dataset["default_filter_string"],
129+
description=dataset["description"],
130+
filter=dataset["filter"], patterns=dataset["patterns"],
131+
target_entity_class=dataset["target_entity_class"])
132+
133+
dataset_cruds = self.cruder.get_all_dataset()
134+
dataset_db_all = self.test_db_session.query(Dataset).all()
135+
# # 断言
136+
self.assertEqual(len(dataset_cruds), len(dataset_db_all))
137+
# # 删除新增的测试数据
138+
dataset_db_instance = self.test_db_session.query(Dataset).get(dataset_id)
139+
self.test_db_session.delete(dataset_db_instance)
140+
self.test_db_session.commit()
141+
142+
@ddt.data(params)
143+
def test_delete_dataset(self, params):
144+
dataset = params["add"]
145+
test_name = util.get_name(dataset["name"])
146+
dataset_id = self.cruder.add_dataset(test_name, datasource_id=dataset["datasource_id"], subid=dataset["subid"],
147+
creator=dataset["creator"], owner=dataset["owner"],
148+
current_process=dataset["current_process"],
149+
last_modifier=dataset["last_modifier"],
150+
data_file_format=dataset["data_file_format"],
151+
default_filter_string=dataset["default_filter_string"],
152+
description=dataset["description"],
153+
filter=dataset["filter"], patterns=dataset["patterns"],
154+
target_entity_class=dataset["target_entity_class"])
155+
156+
self.cruder.delete_dataset(dataset_id)
157+
158+
# 断言
159+
# 根据id查询
160+
dataset_db_instance = self.test_db_session.query(Dataset).get(dataset_id)
161+
self.assertEqual(dataset_db_instance, None)
162+
163+
164+
165+
if __name__ == '__main__':
166+
unittest.main()

0 commit comments

Comments
 (0)