diff --git a/src/recipe_graph/scrape.py b/src/recipe_graph/scrape.py index bdc4519..fd58e82 100644 --- a/src/recipe_graph/scrape.py +++ b/src/recipe_graph/scrape.py @@ -10,7 +10,6 @@ import logging from argparse import ArgumentParser def parse_ingredient(ingredient_text): - units = ['teaspoon', 'tablespoon', 'gram', 'ounce', 'jar', 'cup', 'pinch', 'container', 'slice', 'package', 'pound', 'can', 'dash', 'spear', 'bunch', 'quart', 'cube', 'envelope', 'square', 'sprig', 'bag', diff --git a/test/test_scrape.py b/test/test_scrape.py index e7ac014..eaa84ac 100644 --- a/test/test_scrape.py +++ b/test/test_scrape.py @@ -1,3 +1,12 @@ from recipe_graph import scrape +from bs4 import BeautifulSoup -import pytest \ No newline at end of file +import pytest + + +def test_load_recipe(): + page = scrape.load_recipe("https://hs.andreistoica.ca:4943") + assert type(page) == BeautifulSoup + + page = scrape.load_recipe("https://hs.andreistoica.ca:4943/some-nonesense") + assert page == None \ No newline at end of file