From d404f4c1c33c616b558f4b396a174a07c545e914 Mon Sep 17 00:00:00 2001
From: George Ohashi <george@neuralmagic.com>
Date: Tue, 5 Mar 2024 22:01:55 +0000
Subject: [PATCH] add coverage

---
 Makefile | 7 +++++++
 setup.py | 2 ++
 2 files changed, 9 insertions(+)

diff --git a/Makefile b/Makefile
index 157cbcea8..6b9b4b420 100644
--- a/Makefile
+++ b/Makefile
@@ -78,3 +78,10 @@ clean:
 	rm -rf dist;
 	rm -rf src/sparsezoo.egg-info;
 	find $(PYCHECKDIRS) | grep -E "(__pycache__|\.pyc|\.pyo)" | xargs rm -rf;
+
+
+coverage-report:
+	pytest --cov=sparsezoo --cov-report json
+
+coverage-badge:
+	coverage-badge -f -o ./.meta/coverage_badge.svg
diff --git a/setup.py b/setup.py
index f3cfc958f..5e9b7dfa2 100644
--- a/setup.py
+++ b/setup.py
@@ -63,6 +63,8 @@
     "pytest>=6.0.0",
     "wheel>=0.36.2",
     "matplotlib>=3.0.0",
+    "pytest-cov==4.1.0",
+    "coverage-badge==1.1.0",
 ]
 
 _docs_deps = [