assert_equals: Handle ndarray with booleans

2025-12-06 01:18:52 -06:00 · 2021-05-20 11:38:35 -05:00
parent 52093eb1c0
commit ddd136c661
2 changed files with 8 additions and 1 deletions
--- a/tests/test_features.py
+++ b/tests/test_features.py
@@ -150,5 +150,9 @@ def test_assert_equals() -> None:
        VariableFeatures(values=np.array([1.0, 2.0])),  # type: ignore
        VariableFeatures(values=np.array([1.0, 2.0])),  # type: ignore
    )
+    assert_equals(
+        np.array([True, True]),
+        [True, True],
+    )
    assert_equals((1.0,), (1.0,))
    assert_equals({"x": 10}, {"x": 10})