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