Simulate function for PRG model

simulate.py

from model import Company

def simulate_sample_company(n_steps, **kwargs):
    """
    Runs a model for n_steps and returns a pandas.DataFrame
    containing the data collected at each step.
    kwargs are forwarded to init of company.
    """
    model = Company(**kwargs)
    for i in range(n_steps):
        model.step()
    df = model.data_collector.get_model_vars_dataframe().copy()
    df['i'] = df.index
    df['competency_mechanism'] = model.competency_mechanism
    df['promotion_strategy'] = model.promotion_strategy
    df['year'] = df['i'] * model.timestep_years
    return df