@@ -290,37 +290,40 @@ def test_list_model_versions():
290
290
with mock .patch (
291
291
"sasctl._services.model_repository.ModelRepository.request_link"
292
292
) as request_link :
293
+ with mock .patch (
294
+ "sasctl.core.Session.version_info"
295
+ ) as version :
296
+ version .return_value = VersionInfo (4 )
297
+ get_model_link .return_value = None
298
+ with pytest .raises (ValueError ):
299
+ mr .list_model_versions (
300
+ model = "12345" ,
301
+ )
293
302
294
- get_model_link .return_value = None
295
- with pytest .raises (ValueError ):
296
- mr .list_model_versions (
297
- model = "12345" ,
298
- )
299
-
300
- get_model_link_mock = {
301
- "method" : "GET" ,
302
- "rel" : "modelHistory" ,
303
- "href" : "/modelRepository/models/12345/history" ,
304
- "uri" : "/modelRepository/models/12345/history" ,
305
- "type" : "application/vnd.sas.collection" ,
306
- "responseItemType" : "application/vnd.sas.models.model.version" ,
307
- }
303
+ get_model_link_mock = {
304
+ "method" : "GET" ,
305
+ "rel" : "modelHistory" ,
306
+ "href" : "/modelRepository/models/12345/history" ,
307
+ "uri" : "/modelRepository/models/12345/history" ,
308
+ "type" : "application/vnd.sas.collection" ,
309
+ "responseItemType" : "application/vnd.sas.models.model.version" ,
310
+ }
308
311
309
- get_model_link .return_value = get_model_link_mock
312
+ get_model_link .return_value = get_model_link_mock
310
313
311
- response = mr .list_model_versions (model = "12345" )
312
- assert response
314
+ response = mr .list_model_versions (model = "12345" )
315
+ assert response
313
316
314
- request_link .return_value = RestObj ({"id" : "12345" })
315
- response = mr .list_model_versions (model = "12345" )
316
- assert isinstance (response , list )
317
+ request_link .return_value = RestObj ({"id" : "12345" })
318
+ response = mr .list_model_versions (model = "12345" )
319
+ assert isinstance (response , list )
317
320
318
- request_link .return_value = [
319
- RestObj ({"id" : "12345" }),
320
- RestObj ({"id" : "3456" }),
321
- ]
322
- response = mr .list_model_versions (model = "12345" )
323
- assert isinstance (response , list )
321
+ request_link .return_value = [
322
+ RestObj ({"id" : "12345" }),
323
+ RestObj ({"id" : "3456" }),
324
+ ]
325
+ response = mr .list_model_versions (model = "12345" )
326
+ assert isinstance (response , list )
324
327
325
328
326
329
def test_get_model_version ():
0 commit comments