from mesa import Agent, Model
from mesa.datacollection import DataCollector

class TestAgent(Agent):
    def __init__(self, model):
        super().__init__(model)
        self.wealth = 100
    
    def step(self):
        pass

class TestModel(Model):
    def __init__(self):
        super().__init__()
        agent = TestAgent(self)
        self.add_agent(agent)
    
    def step(self):
        for agent in self.agents:
            agent.step()

model = TestModel()

dc = DataCollector(
    agent_reporters={
        "wealth": "wealth",
        "health": "health",
        "status": "status"
    }
)

dc.collect(model)
print(dc.get_agent_vars_dataframe())