@@ -145,14 +145,14 @@ def test_image_invalid_json_response(self):
145
145
self .project .image (image_id )
146
146
147
147
self .assertIn ("Expecting value" , str (context .exception ))
148
-
148
+
149
149
def test_create_annotation_job_success (self ):
150
150
job_name = "Test Job"
151
151
batch_id = "test-batch-123"
152
152
num_images = 10
153
153
labeler_email = "[email protected] "
154
154
reviewer_email = "[email protected] "
155
-
155
+
156
156
expected_response = {
157
157
"success" : True ,
158
158
"job" : {
@@ -163,68 +163,63 @@ def test_create_annotation_job_success(self):
163
163
"labeler" : labeler_email ,
164
164
"reviewer" : reviewer_email ,
165
165
"status" : "created" ,
166
- "created" : 1616161616
167
- }
166
+ "created" : 1616161616 ,
167
+ },
168
168
}
169
-
169
+
170
170
expected_url = f"{ API_URL } /{ WORKSPACE_NAME } /{ PROJECT_NAME } /jobs?api_key={ ROBOFLOW_API_KEY } "
171
-
171
+
172
172
responses .add (
173
173
responses .POST ,
174
174
expected_url ,
175
175
json = expected_response ,
176
176
status = 200 ,
177
177
match = [
178
- json_params_matcher ({
179
- "name" : job_name ,
180
- "batch" : batch_id ,
181
- "num_images" : num_images ,
182
- "labelerEmail" : labeler_email ,
183
- "reviewerEmail" : reviewer_email
184
- })
185
- ]
178
+ json_params_matcher (
179
+ {
180
+ "name" : job_name ,
181
+ "batch" : batch_id ,
182
+ "num_images" : num_images ,
183
+ "labelerEmail" : labeler_email ,
184
+ "reviewerEmail" : reviewer_email ,
185
+ }
186
+ )
187
+ ],
186
188
)
187
-
189
+
188
190
result = self .project .create_annotation_job (
189
191
name = job_name ,
190
192
batch_id = batch_id ,
191
193
num_images = num_images ,
192
194
labeler_email = labeler_email ,
193
- reviewer_email = reviewer_email
195
+ reviewer_email = reviewer_email ,
194
196
)
195
-
197
+
196
198
self .assertEqual (result , expected_response )
197
199
self .assertTrue (result ["success" ])
198
200
self .assertEqual (result ["job" ]["id" ], "job-123" )
199
201
self .assertEqual (result ["job" ]["name" ], job_name )
200
-
202
+
201
203
def test_create_annotation_job_error (self ):
202
204
job_name = "Test Job"
203
205
batch_id = "invalid-batch"
204
206
num_images = 10
205
207
labeler_email = "[email protected] "
206
208
reviewer_email = "[email protected] "
207
-
208
- error_response = {
209
- "error" : "Batch not found"
210
- }
211
-
209
+
210
+ error_response = {"error" : "Batch not found" }
211
+
212
212
expected_url = f"{ API_URL } /{ WORKSPACE_NAME } /{ PROJECT_NAME } /jobs?api_key={ ROBOFLOW_API_KEY } "
213
-
214
- responses .add (
215
- responses .POST ,
216
- expected_url ,
217
- json = error_response ,
218
- status = 404
219
- )
220
-
213
+
214
+ responses .add (responses .POST , expected_url , json = error_response , status = 404 )
215
+
221
216
with self .assertRaises (RuntimeError ) as context :
222
217
self .project .create_annotation_job (
223
218
name = job_name ,
224
219
batch_id = batch_id ,
225
220
num_images = num_images ,
226
221
labeler_email = labeler_email ,
227
- reviewer_email = reviewer_email
222
+ reviewer_email = reviewer_email ,
228
223
)
229
-
224
+
230
225
self .assertEqual (str (context .exception ), "Batch not found" )
0 commit comments