Bump minor version and test case refactor (#17)

Co-authored-by: zombie-einstein <zombie-einstein@proton.me>
zombie-einstein · Sep 11, 2023 · 7056a47 · 7056a47
1 parent 9c416ac
commit 7056a47
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 4 deletions.
diff --git a/pyproject.toml b/pyproject.toml
@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "jpviz"
-version = "0.1.4"
+version = "0.1.5"
 description = "Jaxpr Visualisation Tool"
 authors = ["zombie-einstein"]
 readme = "README.md"

diff --git a/tests/test_cases.py b/tests/test_cases.py
@@ -90,6 +90,6 @@ def body(carry, a_elems):
 
 
 @pytest.mark.parametrize("f, args", test_cases)
-def test_works(f, args):
-    _ = jpviz.draw(f, collapse_primitives=True)(*args)
-    _ = jpviz.draw(f, collapse_primitives=False)(*args)
+@pytest.mark.parametrize("collapse_primitives", [True, False])
+def test_works(f, args, collapse_primitives):
+    _ = jpviz.draw(f, collapse_primitives=collapse_primitives)(*args)