34
34
ote_eval_testing ,
35
35
ote_train_testing ,
36
36
ote_export_testing ,
37
+ nncf_optimize_testing ,
38
+ nncf_export_testing ,
39
+ nncf_eval_testing ,
40
+ ote_hpo_testing ,
41
+ nncf_eval_openvino_testing ,
42
+ pot_optimize_testing ,
43
+ pot_eval_testing
37
44
)
38
45
39
46
@@ -111,3 +118,45 @@ def test_ote_eval_deployment(self, template):
111
118
@pytest .mark .parametrize ("template" , templates , ids = templates_ids )
112
119
def test_ote_demo_deployment (self , template ):
113
120
ote_demo_deployment_testing (template , root , ote_dir , args )
121
+
122
+ @e2e_pytest_component
123
+ @pytest .mark .parametrize ("template" , templates , ids = templates_ids )
124
+ def test_nncf_optimize (self , template ):
125
+ if template .entrypoints .nncf is None :
126
+ pytest .skip ("nncf entrypoint is none" )
127
+
128
+ nncf_optimize_testing (template , root , ote_dir , args )
129
+
130
+ @e2e_pytest_component
131
+ @pytest .mark .parametrize ("template" , templates , ids = templates_ids )
132
+ def test_nncf_export (self , template ):
133
+ if template .entrypoints .nncf is None :
134
+ pytest .skip ("nncf entrypoint is none" )
135
+
136
+ nncf_export_testing (template , root )
137
+
138
+ @e2e_pytest_component
139
+ @pytest .mark .parametrize ("template" , templates , ids = templates_ids )
140
+ def test_nncf_eval (self , template ):
141
+ if template .entrypoints .nncf is None :
142
+ pytest .skip ("nncf entrypoint is none" )
143
+
144
+ nncf_eval_testing (template , root , ote_dir , args , threshold = 0.001 )
145
+
146
+ @e2e_pytest_component
147
+ @pytest .mark .parametrize ("template" , templates , ids = templates_ids )
148
+ def test_nncf_eval_openvino (self , template ):
149
+ if template .entrypoints .nncf is None :
150
+ pytest .skip ("nncf entrypoint is none" )
151
+
152
+ nncf_eval_openvino_testing (template , root , ote_dir , args )
153
+
154
+ @e2e_pytest_component
155
+ @pytest .mark .parametrize ("template" , templates , ids = templates_ids )
156
+ def test_pot_optimize (self , template ):
157
+ pot_optimize_testing (template , root , ote_dir , args )
158
+
159
+ @e2e_pytest_component
160
+ @pytest .mark .parametrize ("template" , templates , ids = templates_ids )
161
+ def test_pot_eval (self , template ):
162
+ pot_eval_testing (template , root , ote_dir , args )
0 commit comments