From 92b0286af8d48c728921c0eec9c556087eef1e9c Mon Sep 17 00:00:00 2001
From: jbannon <jbann1994@gmail.com>
Date: Fri, 29 Apr 2022 18:00:03 +0000
Subject: [PATCH] src, add unit command

---
 pyproject.toml | 5 +++++
 tools/test     | 2 ++
 2 files changed, 7 insertions(+)

diff --git a/pyproject.toml b/pyproject.toml
index 0b532b09..6589b4a4 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -35,3 +35,8 @@ ignore = [
     "D413", # Missing blank line after last section
     "D415", # Should end with a period
 ]
+
+[tool.coverage.run]
+source = [
+    "src/ytdl_sub"
+]
diff --git a/tools/test b/tools/test
index 16915f28..06ceb188 100755
--- a/tools/test
+++ b/tools/test
@@ -2,6 +2,8 @@
 
 if [[ $1 = "html" ]]; then
     coverage run -m pytest && coverage html
+elif [[ $1 = "unit" ]]; then
+    coverage run -m pytest tests/unit && coverage report
 else
     coverage run -m pytest && coverage report
 fi