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)