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)