diff --git a/tests/nuclear_test.py b/tests/nuclear_test.py index 45093fc..fd9610f 100644 --- a/tests/nuclear_test.py +++ b/tests/nuclear_test.py @@ -113,6 +113,7 @@ def test_varsh_872_5(nuclear: NuclearBogoliubovDrudge): TODO: Investigate its failure in Apache Spark environment. """ + dr = nuclear a, alpha, b, beta, b_prm, beta_prm = symbols( 'a alpha b beta bprm betaprm', integer=True @@ -132,7 +133,7 @@ def test_varsh_872_5(nuclear: NuclearBogoliubovDrudge): ) for sums_i in [sums, reversed(sums)]: tensor = dr.sum(*sums_i, amp) - res = tensor.deep_simplify().merge() + res = tensor.simplify_am().merge() #previously used deep_simplify assert res.n_terms == 1 term = res.local_terms[0] assert len(term.sums) == 0