def test_deserialize_mapped(): s = state.Success(message="1", result=1) f = state.Failed(message="2", result=2) serialized = StateSchema().dump(state.Mapped(message="message", map_states=[s, f])) deserialized = StateSchema().load(serialized) assert isinstance(deserialized, state.Mapped) assert len(deserialized.map_states) == 2 assert deserialized.map_states == [None, None] assert deserialized._result == NoResult
def test_serialize_mapped_uses_set_n_map_states(): serialized = StateSchema().dump(state.Mapped(message="message", n_map_states=20)) assert isinstance(serialized, dict) assert serialized["type"] == "Mapped" assert serialized["message"] == "message" assert "_result" not in serialized assert "map_states" not in serialized assert serialized["n_map_states"] == 20 assert serialized["__version__"] == prefect.__version__
def test_serialize_mapped(): s = state.Success(message="1", result=1) f = state.Failed(message="2", result=2) serialized = StateSchema().dump(state.Mapped(message="message", map_states=[s, f])) assert isinstance(serialized, dict) assert serialized["type"] == "Mapped" assert serialized["message"] == "message" assert "_result" not in serialized assert "map_states" not in serialized assert serialized["n_map_states"] == 2 assert serialized["__version__"] == prefect.__version__
def test_serialize_mapped_uses_set_n_map_states(): serialized = StateSchema().dump(state.Mapped(message="message", n_map_states=20))
serialized = StateSchema().dump(s) assert isinstance(serialized, dict) assert serialized["type"] == "Running" assert serialized["message"] == "hi" assert serialized["__version__"] == prefect.__version__ assert serialized["context"] == s.context deserialized = StateSchema().load(serialized) assert deserialized.is_running() assert deserialized.context == s.context def test_serialize_mapped(): s = state.Success(message="1", result=1) f = state.Failed(message="2", result=2) serialized = StateSchema().dump(state.Mapped(message="message", map_states=[s, f])) assert isinstance(serialized, dict) assert serialized["type"] == "Mapped" assert serialized["message"] == "message" assert "_result" not in serialized assert "map_states" not in serialized assert serialized["n_map_states"] == 2 assert serialized["__version__"] == prefect.__version__ def test_serialize_mapped_uses_set_n_map_states(): serialized = StateSchema().dump(state.Mapped(message="message", n_map_states=20)) assert isinstance(serialized, dict) assert serialized["type"] == "Mapped" assert serialized["message"] == "message" assert "_result" not in serialized