Fix slow basis properties test

opt_einsum is much smarter in calculating this contraction
qutech · Nov 1, 2024 · 333595a · 333595a
1 parent c882d24
commit 333595a
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/tests/test_basis.py b/tests/test_basis.py
@@ -26,6 +26,7 @@
 
 import numpy as np
 import pytest
+from opt_einsum import contract
 from sparse import COO
 
 import filter_functions as ff
@@ -142,7 +143,7 @@ def test_basis_properties(self):
 
             if base.d < 8:
                 # Test very resource intense
-                ref = np.einsum('iab,jbc,kcd,lda', *(base,)*4)
+                ref = contract('iab,jbc,kcd,lda', *(base,)*4)
                 self.assertArrayAlmostEqual(base.four_element_traces.todense(), ref, atol=1e-16)
 
                 # Test setter