wassname/jaxtyping
1
0
Fork 0
mirror of https://github.com/wassname/jaxtyping.git synced 2026-09-10 12:14:04 +08:00
Code Issues Packages Projects Releases Wiki Activity

Broadcast+variadic is now more lenient, and supports anything broadcastable together. Closes #8

Browse Source
This commit is contained in:
Patrick Kidger
2022-07-31 23:59:20 +01:00
parent 6e2837e5b7
commit 6ff4620d1a
3 changed files with 41 additions and 33 deletions
Download Patch File Download Diff File Expand all files Collapse all files
test/test_array.py
+7 -14
View File
@@ -283,12 +283,11 @@ def test_broadcast_variadic_named(typecheck, getkey):
g(b, b)
g(c, c)
g(d, d)
g(b, c)
with pytest.raises(ParamError):
g(a, b)
with pytest.raises(ParamError):
g(a, c)
with pytest.raises(ParamError):
g(b, c)
with pytest.raises(ParamError):
g(a, b)
with pytest.raises(ParamError):
@@ -296,26 +295,20 @@ def test_broadcast_variadic_named(typecheck, getkey):
g(a, j)
g(b, j)
with pytest.raises(ParamError):
g(c, j)
with pytest.raises(ParamError):
g(d, j)
with pytest.raises(ParamError):
g(b, k)
g(c, j)
g(d, j)
g(b, k)
g(c, k)
with pytest.raises(ParamError):
g(d, k)
with pytest.raises(ParamError):
g(c, l)
g(d, l)
with pytest.raises(ParamError):
g(a, m)
g(a, m)
g(c, m)
g(d, m)
with pytest.raises(ParamError):
g(a, n)
with pytest.raises(ParamError):
g(b, n)
g(a, n)
g(b, n)
with pytest.raises(ParamError):
g(c, n)
with pytest.raises(ParamError):