def METHOD_NAME():
    flow = Dataflow()
    flow.flat_map(lambda x: x + 1)
    assert encode_dataflow(flow) == json.dumps(
        {"type": "Dataflow", "steps": [{"type": "FlatMap", "mapper": "<lambda>"}]},
        sort_keys=True,
    )