Skip to content

Commit

Permalink
replace Tuple(items).eval with compile(items)
Browse files Browse the repository at this point in the history
  • Loading branch information
joostvanzwieten committed Apr 23, 2024
1 parent c4fc41f commit bbaa26a
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion tests/test_evaluable.py
Original file line number Diff line number Diff line change
Expand Up @@ -132,7 +132,7 @@ def test_eig(self):
if self.actual.dtype == float:
for ax1, ax2 in self.pairs:
items = self.actual, *evaluable.eig(self.actual, axes=(ax1, ax2))
A, L, V = evaluable.Tuple(items).simplified.eval(**dict(zip(self.arg_names, self.arg_values)))
A, L, V = evaluable.compile(items, simplify=True, stats=False)(**dict(zip(self.arg_names, self.arg_values)))
self.assertArrayAlmostEqual(decimal=11,
actual=(numpy.expand_dims(V, ax2) * numpy.expand_dims(L, ax2+1).swapaxes(ax1, ax2+1)).sum(ax2+1),
desired=(numpy.expand_dims(A, ax2) * numpy.expand_dims(V, ax2+1).swapaxes(ax1, ax2+1)).sum(ax2+1))
Expand Down

0 comments on commit bbaa26a

Please sign in to comment.