def collect_widgets(self): """Returns all the widgets the Collector wants.""" return [widget.Widget() for _ in xrange(self.desired_widget_count)]
def testWidgetBang(self): toy = widget.Widget() self.assertEqual(toy.bang(), 'bang bang!') self.assertEqual(toy.bang('boom'), 'boom bang!')
def __init__(self): self.widget = widget.Widget() self.desired_widget_count = 10
def testWidgetWhack(self): toy = widget.Widget() self.assertEqual(toy.whack(), 'whack!') self.assertEqual(toy.whack(3), 'whack! whack! whack!')