def test_ml_messge_consistency():
"""Test the consistency of the message."""
dm = DataModel("ml_datamodel", [Attribute("dataset_id", str, True)])
query = Query([Constraint("dataset_id", ConstraintType("==", "fmnist"))], model=dm)
msg = MlTradeMessage(performative=MlTradeMessage.Performative.CFP, query=query)
with mock.patch.object(MlTradeMessage.Performative, "__eq__", return_value=False):
assert not msg._is_consistent()
def test_ml_wrong_message_creation():
"""Test the creation of a ml message."""
msg = MlTradeMessage(performative=MlTradeMessage.Performative.CFP, query="")
assert not msg._is_consistent()