def test_set_docstring_blank(self):
this = os.path.dirname(__file__)
onnx_file = os.path.join(this, "models", "coreml_OneHotEncoder_BikeSharing.onnx")
onnx_model = load_model(onnx_file)
set_model_doc_string(onnx_model, "sample")
self.assertRaises(ValueError, set_model_doc_string, onnx_model.doc_string, "sample")
set_model_doc_string(onnx_model, "", True)
self.assertEqual(onnx_model.doc_string, "")
def test_model_setters(self):
this = os.path.dirname(__file__)
onnx_file = os.path.join(this, "models", "coreml_OneHotEncoder_BikeSharing.onnx")
onnx_model = load_model(onnx_file)
set_model_version(onnx_model, 2)
set_model_domain(onnx_model, "com.sample")
set_model_doc_string(onnx_model, "Sample docstring")
self.assertEqual(onnx_model.model_version, 2)
self.assertEqual(onnx_model.domain, "com.sample")
self.assertEqual(onnx_model.doc_string, "Sample docstring")
