def METHOD_NAME(patch_generate_cohort):
    main(
        [
            "generate_cohort",
            "--study-definition",
            "study_definition",
            "--output-file",
            "mydir/input.feather",
            "--output-dir",
            "mydir",
            "--output-format",
            "feather",
        ]
    )
    patch_generate_cohort.assert_called_once()