test: fix

Signed-off-by: nstarman <nstarman@users.noreply.github.com>
GalacticDynamics · Oct 22, 2024 · 624c282 · 624c282
1 parent 5468fcd
commit 624c282
Show file tree

Hide file tree

Showing 2 changed files with 27 additions and 18 deletions.
diff --git a/tests/unit/test_quantity.py b/tests/unit/test_quantity.py
@@ -233,7 +233,9 @@ def test_gt():
     assert jnp.array_equal(q > Quantity(1, u.m), jnp.array([False, True, True]))
 
     # Test with incompatible units
-    assert jnp.array_equal(q > Quantity(0, u.s), jnp.array([False, False, False]))
+    # TODO: better equinox exception matching
+    with pytest.raises(Exception):  # noqa: B017, PT011
+        _ = q > Quantity(0, u.s)
 
 
 def test_ge():
@@ -251,7 +253,9 @@ def test_ge():
     assert jnp.array_equal(q >= Quantity(2, u.m), jnp.array([False, True, True]))
 
     # Test with incompatible units
-    assert jnp.array_equal(q >= Quantity(0, u.s), jnp.array([False, False, False]))
+    # TODO: better equinox exception matching
+    with pytest.raises(Exception):  # noqa: B017, PT011
+        _ = q >= Quantity(0, u.s)
 
 
 def test_lt():
@@ -269,7 +273,9 @@ def test_lt():
     assert jnp.array_equal(q < Quantity(2, u.m), jnp.array([True, False, False]))
 
     # Test with incompatible units
-    assert jnp.array_equal(q < Quantity(0, u.s), jnp.array([False, False, False]))
+    # TODO: better equinox exception matching
+    with pytest.raises(Exception):  # noqa: B017, PT011
+        _ = q < Quantity(0, u.s)
 
 
 def test_le():
@@ -287,7 +293,9 @@ def test_le():
     assert jnp.array_equal(q <= Quantity(2, u.m), jnp.array([True, True, False]))
 
     # Test with incompatible units
-    assert jnp.array_equal(q <= Quantity(0, u.s), jnp.array([False, False, False]))
+    # TODO: better equinox exception matching
+    with pytest.raises(Exception):  # noqa: B017, PT011
+        _ = q <= Quantity(0, u.s)
 
 
 def test_eq():
@@ -305,7 +313,9 @@ def test_eq():
     assert jnp.array_equal(q == Quantity(2, u.m), jnp.array([False, True, False]))
 
     # Test with incompatible units
-    assert jnp.array_equal(q == Quantity(0, u.s), jnp.array([False, False, False]))
+    # TODO: better equinox exception matching
+    with pytest.raises(Exception):  # noqa: B017, PT011
+        _ = q == Quantity(0, u.s)
 
 
 def test_ne():
@@ -323,8 +333,11 @@ def test_ne():
     assert jnp.array_equal(q != Quantity(2, u.m), jnp.array([True, False, True]))
 
     # Test with incompatible units
-    assert jnp.array_equal(q != Quantity(0, u.s), jnp.array([True, True, True]))
-    assert jnp.array_equal(q != Quantity(4, u.s), jnp.array([True, True, True]))
+    # TODO: better equinox exception matching
+    with pytest.raises(Exception):  # noqa: B017, PT011
+        _ = q != Quantity(0, u.s)
+    with pytest.raises(Exception):  # noqa: B017, PT011
+        _ = q != Quantity(4, u.s)
 
 
 def test_neg():

diff --git a/uv.lock b/uv.lock