def add_item_stash(scene):
""" Add an item stash to a scene."""
# Create pool of items to choose from
# 1. Take 2 weapons, put in pool
pool = []
for i in range(2):
pool.append(items.new_weapon())
# 2. Take 1 armor, put in pool
for i in range(1):
pool.append(items.new_armor())
# 3. Randomly pick 1-2 items from pool
shuffle(pool)
final = []
for i in range(randint(1, 2)):
final.append(pool.pop())
# Create item stash and add to scene
stash = map_.ItemStash(final)
scene.features.append(stash)
def new_armor_test():
# Tests if new armors are created properly.
armor = items.new_armor()
ok_(armor is not None)
