diff --git a/my_flask_app/commands.py b/my_flask_app/commands.py index 6c21d07..1182904 100644 --- a/my_flask_app/commands.py +++ b/my_flask_app/commands.py @@ -19,14 +19,22 @@ is_flag=True, help="Show coverage report", ) -def test(coverage): +@click.option( + "-k", + "--filter", + default=None, + help="Filter tests by keyword expressions", +) +def test(coverage, filter): """Run the tests.""" import pytest args = [TEST_PATH, "--verbose"] if coverage: args.append("--cov=my_flask_app") - rv = pytest.main(args) + if filter: + args.extend(["-k", filter]) + rv = pytest.main(args=args) exit(rv)