class InfoTest(VKBaseTest): def setUp(self): super(InfoTest, self).setUp() self.info = Info(self.items) self.gems = self.info.get_good('gems') def test_price(self): self.eq(self.info.price(self.gems, 3), 1) self.eq(self.info.price(self.gems, 4), 2) def test_title(self): self.eq(self.info.title('gems', 4), u'20 алмазов') self.eq(self.info.title('gems', 3), u'10 алмазов') self.eq(self.info.title('gems', 1), u'1 алмаз') self.eq(self.info.title('gems', 2), u'3 алмаза') def test_info(self): self.eq( self.info('gems', 3), self.info.response( { 'title': self.info.title('gems', 3), 'price': 1, 'photo_url': self.items['gems'][3]['image'] } ) ) def test_unknown_item(self): self.raises(UnknownItemError, self.info.get_good, 'item') def test_invalid_count(self): self.raises(InvalidCountError, self.info.price, self.gems, 11)
def setUp(self): super(InfoTest, self).setUp() self.info = Info(self.items) self.gems = self.info.get_good('gems')