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")