From 294231dd48e1df1f8d5956ddb5361fd95b16cf5d Mon Sep 17 00:00:00 2001 From: Andrei Stoica Date: Sat, 15 Oct 2022 14:33:45 -0400 Subject: [PATCH] added test exclusion to main functions --- src/recipe_graph/db.py | 4 ++-- src/recipe_graph/insert_sites.py | 4 ++-- 2 files changed, 4 insertions(+), 4 deletions(-) diff --git a/src/recipe_graph/db.py b/src/recipe_graph/db.py index 73ccd6c..8dc7ad6 100644 --- a/src/recipe_graph/db.py +++ b/src/recipe_graph/db.py @@ -250,10 +250,10 @@ def update_graph_connectivity(session=None): session.add(RecipeGraphed(recipe_id=recipe.id, status=True)) -def main(): +def main(): # pragma: no cover eng = get_engine() create_tables(eng) -if __name__ == "__main__": +if __name__ == "__main__": # pragma: no cover main() diff --git a/src/recipe_graph/insert_sites.py b/src/recipe_graph/insert_sites.py index a4a1c2b..812bd76 100644 --- a/src/recipe_graph/insert_sites.py +++ b/src/recipe_graph/insert_sites.py @@ -29,7 +29,7 @@ def setup_logging(args: argparse.Namespace) -> logging.Logger: return logger -def main(): +def main(): # pragma: no cover args = setup_argparser(sys.argv[1:]) logger = setup_logging(args) @@ -42,5 +42,5 @@ def main(): session.add(db.RecipeSite(**site)) -if __name__ == "__main__": +if __name__ == "__main__": # pragma: no cover main()