update scripts and makefile

Author: tylerhutcherson

Makefile

@@ -1,4 +1,4 @@
-.PHONY: install format lint test clean redis-start redis-stop check-types integration-test docs-build docs-serve check
+.PHONY: install format lint test test-all clean redis-start redis-stop check-types docs-build docs-serve check
 
 install:
 	poetry install --all-extras

scripts.py

@@ -1,4 +1,5 @@
 import subprocess
+import sys
 
 
 def format():
@@ -29,13 +30,23 @@ def check_mypy():
 
 
 def test():
-    subprocess.run(["python", "-m", "pytest", "-n", "auto", "--log-level=CRITICAL"], check=True)
+    # Get any extra arguments passed to the script
+    extra_args = sys.argv[1:]
+    if not extra_args:
+        test_cmd = ["python", "-m", "pytest", "-n", "auto", "--log-level=CRITICAL"]
+    else:
+        test_cmd = ["python", "-m", "pytest", "-n", "auto", "--log-level=CRITICAL"] + extra_args
+    subprocess.run(test_cmd, check=True)
 
 
 def test_verbose():
-    subprocess.run(
-        ["python", "-m", "pytest", "-n", "auto", "-vv", "-s", "--log-level=CRITICAL"], check=True
-    )
+    # Get any extra arguments passed to the script
+    extra_args = sys.argv[1:]
+    if not extra_args:
+        test_cmd = ["python", "-m", "pytest", "-n", "auto", "-vv", "-s", "--log-level=CRITICAL"]
+    else:
+        test_cmd = ["python", "-m", "pytest", "-n", "auto", "-vv", "-s", "--log-level=CRITICAL"] + extra_args
+    subprocess.run(test_cmd, check=True)
 
 
 def test_notebooks():