From b6daacca2d2aa203316b8668a1ec95c8656eddb7 Mon Sep 17 00:00:00 2001 From: Andrei Stoica Date: Wed, 17 May 2023 22:01:19 -0400 Subject: [PATCH] test load_recipe --- src/recipe_graph/scrape.py | 1 - test/test_scrape.py | 11 ++++++++++- 2 files changed, 10 insertions(+), 2 deletions(-) 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