def make_provider(name, needs=None, provides=None, cost=50): if not needs: needs = [] if not provides: provides = [] provider = BaseProvider() provider.name = name provider.get_provides = Mock(return_value=provides) provider.get_needs = Mock(return_value=needs) provider.get_cost = Mock(return_value=cost) return provider