added tests and refactoring

This commit is contained in:
Samuel Puček 2022-08-06 19:10:32 +02:00
parent bc7de77181
commit 83750d99a6
4 changed files with 169 additions and 43 deletions

2
.gitignore vendored
View File

@ -5,3 +5,5 @@ obj
vendor vendor
.idea .idea
*.iml *.iml
venv
settings.json

View File

@ -1,39 +1,56 @@
# -*- coding: utf-8 -*- # -*- coding: utf-8 -*-
class GildedRose(object):
class GildedRose(object):
def __init__(self, items): def __init__(self, items):
self.items = items self.items = items
def update_quality(self): def update_quality(self):
for item in self.items: for item in self.items:
if item.name != "Aged Brie" and item.name != "Backstage passes to a TAFKAL80ETC concert":
if item.quality > 0: if item.name == "Sulfuras, Hand of Ragnaros":
if item.name != "Sulfuras, Hand of Ragnaros": # Sulfuras has quality 80, sell in value does not update
item.quality = item.quality - 1 if item.quality != 80:
else: raise ValueError(
if item.quality < 50: "Sulfuras, Hand of Ragnaros has to have a quality equal 80." # noqa: E501
item.quality = item.quality + 1 )
if item.name == "Backstage passes to a TAFKAL80ETC concert":
if item.sell_in < 11: elif item.name == "Aged Brie":
if item.quality < 50: if item.sell_in > 0:
item.quality = item.quality + 1 item.quality += 1
if item.sell_in < 6:
if item.quality < 50:
item.quality = item.quality + 1
if item.name != "Sulfuras, Hand of Ragnaros":
item.sell_in = item.sell_in - 1
if item.sell_in < 0:
if item.name != "Aged Brie":
if item.name != "Backstage passes to a TAFKAL80ETC concert":
if item.quality > 0:
if item.name != "Sulfuras, Hand of Ragnaros":
item.quality = item.quality - 1
else:
item.quality = item.quality - item.quality
else: else:
if item.quality < 50: item.quality += 2
item.quality = item.quality + 1 item.quality = min(item.quality, 50)
item.sell_in -= 1
elif item.name == "Backstage passes to a TAFKAL80ETC concert":
if item.sell_in <= 0:
item.quality = 0
elif item.sell_in <= 5:
item.quality += 3
elif item.sell_in <= 10:
item.quality += 2
else:
item.quality += 1
item.quality = min(item.quality, 50)
item.sell_in -= 1
elif "conjured" in item.name.lower():
if item.sell_in > 0:
item.quality -= 2
else:
item.quality -= 4
item.quality = max(item.quality, 0)
item.sell_in -= 1
else:
# Standard goods
if item.sell_in > 0:
item.quality -= 1
else:
item.quality -= 2
item.sell_in -= 1
item.quality = max(item.quality, 0)
class Item: class Item:

View File

@ -9,8 +9,108 @@ class GildedRoseTest(unittest.TestCase):
items = [Item("foo", 0, 0)] items = [Item("foo", 0, 0)]
gilded_rose = GildedRose(items) gilded_rose = GildedRose(items)
gilded_rose.update_quality() gilded_rose.update_quality()
self.assertEquals("fixme", items[0].name) self.assertEquals("foo", items[0].name)
def test_sulfuras_constant_quality_and_sell_in_value_update(self):
if __name__ == '__main__': items = [Item("Sulfuras, Hand of Ragnaros", 20, 80)]
gilded_rose = GildedRose(items)
gilded_rose.update_quality()
self.assertEqual(80, items[0].quality)
self.assertEqual(20, items[0].sell_in)
def test_normal_goods_fresh_quality_and_sell_in_value_update(self):
items = [Item("Rat Soup", 20, 10)]
gilded_rose = GildedRose(items)
gilded_rose.update_quality()
self.assertEqual(9, items[0].quality)
self.assertEqual(19, items[0].sell_in)
def test_normal_goods_rotten_quality_and_sell_in_value_update(self):
items = [Item("Rat Soup", 0, 5)]
gilded_rose = GildedRose(items)
gilded_rose.update_quality()
self.assertEqual(3, items[0].quality)
self.assertEqual(-1, items[0].sell_in)
def test_normal_goods_non_negative_quality(self):
items = [Item("Rat Soup light", 0, 0)]
gilded_rose = GildedRose(items)
gilded_rose.update_quality()
self.assertEqual(0, items[0].quality)
self.assertEqual(-1, items[0].sell_in)
def test_aged_brie_quality_and_sell_in_value_standard_update(self):
items = [Item("Aged Brie", 20, 30)]
gilded_rose = GildedRose(items)
gilded_rose.update_quality()
self.assertEqual(31, items[0].quality)
self.assertEqual(19, items[0].sell_in)
def test_aged_brie_quality_and_sell_in_value_double_update(self):
items = [Item("Aged Brie", 0, 30)]
gilded_rose = GildedRose(items)
gilded_rose.update_quality()
self.assertEqual(32, items[0].quality)
self.assertEqual(-1, items[0].sell_in)
def test_aged_brie_quality_hit_50(self):
items = [Item("Aged Brie", -5, 49)]
gilded_rose = GildedRose(items)
gilded_rose.update_quality()
self.assertEqual(50, items[0].quality)
self.assertEqual(-6, items[0].sell_in)
def test_backstage_passes_quality_and_sell_in_value_standard_update(self):
items = [Item("Backstage passes to a TAFKAL80ETC concert", 20, 10)]
gilded_rose = GildedRose(items)
gilded_rose.update_quality()
self.assertEqual(11, items[0].quality)
self.assertEqual(19, items[0].sell_in)
def test_backstage_passes_quality_and_sell_in_value_double_update(self):
items = [Item("Backstage passes to a TAFKAL80ETC concert", 8, 10)]
gilded_rose = GildedRose(items)
gilded_rose.update_quality()
self.assertEqual(12, items[0].quality)
self.assertEqual(7, items[0].sell_in)
def test_backstage_passes_quality_and_sell_in_value_triple_update(self):
items = [Item("Backstage passes to a TAFKAL80ETC concert", 4, 10)]
gilded_rose = GildedRose(items)
gilded_rose.update_quality()
self.assertEqual(13, items[0].quality)
self.assertEqual(3, items[0].sell_in)
def test_backstage_passes_quality_and_sell_in_value_after_concert_update(
self,
): # noqa: E501
items = [Item("Backstage passes to a TAFKAL80ETC concert", 0, 10)]
gilded_rose = GildedRose(items)
gilded_rose.update_quality()
self.assertEqual(0, items[0].quality)
self.assertEqual(-1, items[0].sell_in)
def test_conjured_items_quality_and_sell_in_value_standard_update(self):
items = [Item("Conjured Pizza", 5, 10)]
gilded_rose = GildedRose(items)
gilded_rose.update_quality()
self.assertEqual(8, items[0].quality)
self.assertEqual(4, items[0].sell_in)
def test_conjured_items_quality_and_sell_in_value_rotten_update(self):
items = [Item("Conjured Pizza", 0, 10)]
gilded_rose = GildedRose(items)
gilded_rose.update_quality()
self.assertEqual(6, items[0].quality)
self.assertEqual(-1, items[0].sell_in)
def test_conjured_items_quality_hit_0(self):
items = [Item("Conjured Pizza", 0, 1)]
gilded_rose = GildedRose(items)
gilded_rose.update_quality()
self.assertEqual(0, items[0].quality)
self.assertEqual(-1, items[0].sell_in)
if __name__ == "__main__":
unittest.main() unittest.main()

View File

@ -1,24 +1,31 @@
# -*- coding: utf-8 -*- # -*- coding: utf-8 -*-
import sys
from __future__ import print_function from __future__ import print_function
from gilded_rose import Item, GildedRose
from gilded_rose import *
if __name__ == "__main__": if __name__ == "__main__":
print ("OMGHAI!") print("OMGHAI!")
items = [ items = [
Item(name="+5 Dexterity Vest", sell_in=10, quality=20), Item(name="+5 Dexterity Vest", sell_in=10, quality=20),
Item(name="Aged Brie", sell_in=2, quality=0), Item(name="Aged Brie", sell_in=2, quality=0),
Item(name="Elixir of the Mongoose", sell_in=5, quality=7), Item(name="Elixir of the Mongoose", sell_in=5, quality=7),
Item(name="Sulfuras, Hand of Ragnaros", sell_in=0, quality=80), Item(name="Sulfuras, Hand of Ragnaros", sell_in=0, quality=80),
Item(name="Sulfuras, Hand of Ragnaros", sell_in=-1, quality=80), Item(name="Sulfuras, Hand of Ragnaros", sell_in=-1, quality=80),
Item(name="Backstage passes to a TAFKAL80ETC concert", sell_in=15, quality=20), Item(
Item(name="Backstage passes to a TAFKAL80ETC concert", sell_in=10, quality=49), name="Backstage passes to a TAFKAL80ETC concert", sell_in=15, quality=20
Item(name="Backstage passes to a TAFKAL80ETC concert", sell_in=5, quality=49), ), # noqa: E501
Item(name="Conjured Mana Cake", sell_in=3, quality=6), # <-- :O Item(
] name="Backstage passes to a TAFKAL80ETC concert", sell_in=10, quality=49
), # noqa: E501
Item(
name="Backstage passes to a TAFKAL80ETC concert", sell_in=5, quality=49
), # noqa: E501
Item(name="Conjured Mana Cake", sell_in=3, quality=6), # <-- :O
]
days = 2 days = 2
import sys
if len(sys.argv) > 1: if len(sys.argv) > 1:
days = int(sys.argv[1]) + 1 days = int(sys.argv[1]) + 1
for day in range(days): for day in range(days):