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 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
