Add hypothesis

nerfstudio-project · Sep 16, 2024 · 2e7765d · 2e7765d
1 parent 9b7c19d
commit 2e7765d
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 4 deletions.
diff --git a/pyproject.toml b/pyproject.toml
@@ -43,6 +43,7 @@ dev = [
     "ruff==0.6.2",
     "pre-commit==3.3.2",
     "pytest",
+    "hypothesis[numpy]",
 ]
 examples = [
     "torch>=1.13.1",

diff --git a/tests/test_transforms_ops.py b/tests/test_transforms_ops.py
@@ -4,16 +4,15 @@
 
 import numpy as onp
 import numpy.typing as onpt
-from jax import numpy as jnp
+import viser.transforms as vtf
+
 from utils import (
     assert_arrays_close,
     assert_transforms_close,
     general_group_test,
     sample_transform,
 )
 
-import viser.transforms as vtf
-
 
 @general_group_test
 def test_sample_uniform_valid(
@@ -119,7 +118,7 @@ def test_multiply(
     T_b_a = sample_transform(Group, batch_axes, dtype)
     assert_arrays_close(
         onp.einsum(
-            "...ij,...jk->...ik", T_w_b.as_matrix(), jnp.linalg.inv(T_w_b.as_matrix())
+            "...ij,...jk->...ik", T_w_b.as_matrix(), onp.linalg.inv(T_w_b.as_matrix())
         ),
         onp.broadcast_to(
             onp.eye(Group.matrix_dim, dtype=dtype),