def test_agents(agent_type, attributes): agents = testing.agents(size_strategy=st.just(10), agent_type=agent_type, attributes=attributes) assert agents.example().shape == (10,)