def test_with_return_true(self): assert ModelExportUtils.file_filter('abc.mxnet', self.files_to_exclude) is True
def test_with_pyc(self): assert ModelExportUtils.file_filter('abc.pyc', self.files_to_exclude) is False
def test_with_ds_store(self): assert ModelExportUtils.file_filter('.DS_Store', self.files_to_exclude) is False
def test_with_return_false(self): assert ModelExportUtils.file_filter('abc.onnx', self.files_to_exclude) is False