def test_create_simple_transit_subsidy(self): trans = self._set_transit_subsidy() trans.save() # print self.modes #Metro _modes = TransitSubsidyModes(transit_subsidy=trans, mode=self.modes[0], cost=100) _modes.save() #Dash _modes = TransitSubsidyModes(transit_subsidy=trans, mode=self.modes[1], cost=50) _modes.save() #Other _modes = TransitSubsidyModes(transit_subsidy=trans, mode=self.modes[4], cost=5, other_mode='ScooterBus') _modes.save() #Guard self.assertTrue( _modes.timestamp != None) ts_modes = TransitSubsidyModes.objects.all() actual = ts_modes.aggregate( Sum('cost') ) print actual['cost__sum'] self.assertEquals( Decimal('155.00'), actual['cost__sum'] ) trans2 = TransitSubsidy.objects.filter(user=self.user)[0] self.assertEquals('patti smith', trans2.signature)
def set_modes(self): #Metro self.modes = Mode.objects.all() trans = self._set_transit_subsidy() _modes = TransitSubsidyModes(transit_subsidy=trans, mode=self.modes[0], cost=100) _modes.save() #Dash _modes = TransitSubsidyModes(transit_subsidy=trans, mode=self.modes[1], cost=50) _modes.save() #Other _modes = TransitSubsidyModes(transit_subsidy=trans, mode=self.modes[4], cost=5, other_mode='ScooterBus') _modes.save()
def test_get_modes_as_string(self): trans = self._set_transit_subsidy() trans.save() # print self.modes #Metro _modes = TransitSubsidyModes(transit_subsidy=trans, mode=self.modes[0], cost=100) _modes.save() #Dash _modes = TransitSubsidyModes(transit_subsidy=trans, mode=self.modes[1], cost=50) _modes.save() #Other _modes = TransitSubsidyModes(transit_subsidy=trans, mode=self.modes[4], cost=5, other_mode='ScooterBus') _modes.save() self.assertTrue( self.modes[0].__unicode__() != None ) self.assertTrue( _modes.__unicode__() != None )