def output_df(spark):
    """
    This is the expected result of the function to be tested.
    
    Defined as a fixture, in the same way as the input_df
    """
    return (spark.createDataFrame([
        ["Cat", 4],
        ["Hamster", 1],
        ["Dog", 2],
        ], ["animal_group", "count"]))