Python ModelExportUtils.find_unique Examples

Example #1

File: test_model_packaging_utils.py Project: rizvi-kth/Deeplearn

 def test_with_count_zero(self):
     files = ['a.txt', 'b.txt', 'c.txt']
     suffix = '.mxnet'
     val = ModelExportUtils.find_unique(files, suffix)
     assert val is None

Example #2

File: test_model_packaging_utils.py Project: rizvi-kth/Deeplearn

 def test_with_exit(self):
     files = ['a.onnx', 'b.onnx', 'c.txt']
     suffix = '.onnx'
     with pytest.raises(ModelArchiverError):
         ModelExportUtils.find_unique(files, suffix)