def test_version_exists(self):
model = ModelProto()
# When we create it, graph should not have a version string.
self.assertFalse(model.HasField('ir_version'))
# We should touch the version so it is annotated with the current
# ir version of the running ONNX
model.ir_version = IR_VERSION
model_string = model.SerializeToString()
model.ParseFromString(model_string)
self.assertTrue(model.HasField('ir_version'))
# Check if the version is correct.
self.assertEqual(model.ir_version, IR_VERSION)
loaded_model = onnx.load(f)
assert model == loaded_model
# Test if input is a file name
f = tempfile.NamedTemporaryFile(delete=False)
f.write(model_string)
f.close()
loaded_model = onnx.load(f.name)
assert model == loaded_model
os.remove(f.name)
try:
AttributeProto
NodeProto
GraphProto
ModelProto
except Exception as e:
assert False, 'Did not find proper onnx protobufs. Error is: {}'.format(e)
model = ModelProto()
# When we create it, graph should not have a version string.
assert not model.HasField('ir_version')
# We should touch the version so it is annotated with the current
# ir version of the running ONNX
model.ir_version = IR_VERSION
model_string = model.SerializeToString()
model.ParseFromString(model_string)
assert model.HasField('ir_version')
# Check if the version is correct.
assert model.ir_version == IR_VERSION
