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
def test_with_exit(self):
files = ['a.onnx', 'b.onnx', 'c.txt']
suffix = '.onnx'
with pytest.raises(ModelArchiverError):
ModelExportUtils.find_unique(files, suffix)