async def METHOD_NAME(self):
    filename = self.mktempfile() + ".json"
    pathlib.Path(filename).write_text(json.dumps(self.DATAFLOW.export()))
    with contextlib.redirect_stdout(self.stdout):
        await CLI.cli(
            "dataflow",
            "diagram",
            "-stages",
            "processing",
            "output",
            "--",
            filename,
        )
    stdout = self.stdout.getvalue()
    # Check that the single stage is not its own subgraph
    for check in ["Processing", "Output"]:
        self.assertIn(f"{check} Stage", stdout)
    for check in ["Cleanup"]:
        self.assertNotIn(f"{check} Stage", stdout)