Skip to content

Commit 080f7ca

Browse files
nit
1 parent d1ae49b commit 080f7ca

1 file changed

Lines changed: 4 additions & 2 deletions

File tree

test/test_value_array.py

Lines changed: 4 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -298,5 +298,7 @@ def test_unique() -> None:
298298
def test_matmul() -> None:
299299
a = np.random.random((3, 4))
300300
b = np.random.random((4, 3)) * tu.ns
301-
assert (a @ b).allclose((a @ b[tu.us]) * tu.us)
302-
assert (b @ a).allclose((b[tu.s] @ a) * tu.s)
301+
c: tu.TimeArray = a @ b # type: ignore[assignment]
302+
d: tu.TimeArray = b @ a # type: ignore[assignment]
303+
assert c.allclose((a @ b[tu.us]) * tu.us)
304+
assert d.allclose((b[tu.s] @ a) * tu.s)

0 commit comments

Comments
 (0)