Skip to content

Commit

Permalink
Add test
Browse files Browse the repository at this point in the history
  • Loading branch information
fferflo committed Jun 11, 2024
1 parent 5af3540 commit 254797d
Showing 1 changed file with 9 additions and 1 deletion.
10 changes: 9 additions & 1 deletion test/test_compare.py
Original file line number Diff line number Diff line change
Expand Up @@ -424,6 +424,14 @@ def test_compare_numpy():
einx.flip("a [b]", x),
)

# np.flipud
x = np.random.uniform(size=(2, 3))
assert_equal_numpy(
np.flipud(x),
einx.flip("[a] b", x),
)


if importlib.util.find_spec("scipy"):
import numpy as np
import scipy.linalg
Expand All @@ -442,4 +450,4 @@ def test_compare_scipy():
assert_equal_numpy(
scipy.linalg.khatri_rao(x, y),
einx.multiply("a c, b c -> (a b) c", x, y),
)
)

0 comments on commit 254797d

Please sign in to comment.