def test_can_add_items_to_category_with_existing_items(): item_1 = Item('Leinenkugel Creamy Dark Lager', 'Chippewa Falls, WI', 4.00) item_2 = Item('Budweiser', 'St. Louis, MO', 3.50) item_3 = Item('PBR', 'Some really obscure place you probably never heard of', 3.25) category = Category("Beer", [item_1, item_2]) category.add_item(item_3) assert item_1 in category.items assert item_2 in category.items assert item_3 in category.items
def populateDB(restaurants): for name, properties in restaurants.iteritems(): # Extract dictionary address = ", ".join(properties['address']) description = properties['description'] phone = properties['phone'] website = properties['website'] email = properties['email'] loc = (float(properties['longitude']), float(properties['latitude'])) # get image name only img = properties['image'].split('/')[-1].strip() categories = properties['categories'] try: # Only add new restaurant Restaurant.objects.all().get(name=name) except: # Create object in database r = Restaurant(name=name, address=address, description=description, phone=phone, website=website, email=email, img=img, pt=Point(loc)) r.save() print r i = 0 while i < len(categories) and i < 2: catName = categories[i] catName = catName.upper() c = None try: # Get existing category if exists c = Category.objects.get(name=catName) except: # Create new category if i == 0: c = Category(name=catName) # First cat else: c = Category(name=catName, super_cat=categories[0].upper()) # Sub cat c.save() print c r.categories.add(c) r.save() i += 1
def test_two_categories_have_separate_items(): category_1 = Category("Beer") category_2 = Category("Appetizers") beer = Item('Leinenkugel Creamy Dark Lager', 'Chippewa Falls, WI', 4.00) quesadillas = Item( 'Quesadillas', 'Steak or chicken in a grilled flour tortilla with melted cheese', 8.25) category_1.add_item(beer) category_2.add_item(quesadillas) assert beer in category_1.items assert quesadillas in category_2.items assert beer not in category_2.items assert quesadillas not in category_1.items
def test_two_categories_have_separate_items(): category_1 = Category("Beer") category_2 = Category("Appetizers") beer = Item('Leinenkugel Creamy Dark Lager', 'Chippewa Falls, WI', 4.00) quesadillas = Item('Quesadillas', 'Steak or chicken in a grilled flour tortilla with melted cheese', 8.25) category_1.add_item(beer) category_2.add_item(quesadillas) assert beer in category_1.items assert quesadillas in category_2.items assert beer not in category_2.items assert quesadillas not in category_1.items
def test_item_added_to_empty_category_exists(): category = Category("Beer") item = Item('Leinenkugel Creamy Dark Lager', 'Chippewa Falls, WI', 4.00) category.add_item(item); assert item in category.items
def test_item_added_to_empty_category_exists(): category = Category("Beer") item = Item('Leinenkugel Creamy Dark Lager', 'Chippewa Falls, WI', 4.00) category.add_item(item) assert item in category.items