change: improve unit tests for jumpstart cache

evakravi · evakravi · commit 37a502eff175 · 2021-11-11T18:40:45.000Z
diff --git a/tests/unit/sagemaker/jumpstart/test_cache.py b/tests/unit/sagemaker/jumpstart/test_cache.py
@@ -335,6 +335,7 @@ def test_jumpstart_cache_accepts_input_parameters():
         manifest_file_s3_key=manifest_file_key,
     )
 
+    assert cache.get_manifest_file_s3_key() == manifest_file_key
     assert cache.get_region() == region
     assert cache.get_bucket() == bucket
     assert cache._s3_cache._max_cache_items == max_s3_cache_items

Original file line number	Diff line number	Diff line change
`@@ -335,6 +335,7 @@ def test_jumpstart_cache_accepts_input_parameters():`
`335`	`335`	`manifest_file_s3_key=manifest_file_key,`
`336`	`336`	`)`
`337`	`337`
	`338`	`+ assert cache.get_manifest_file_s3_key() == manifest_file_key`
`338`	`339`	`assert cache.get_region() == region`
`339`	`340`	`assert cache.get_bucket() == bucket`
`340`	`341`	`assert cache._s3_cache._max_cache_items == max_s3_cache_items`