Recursively evaluate combined factors

Author: asterycs

src/forward.jl

@@ -269,6 +269,9 @@ function evaluate(::Mult, arg1::BinaryOperation{Mult}, arg2::BinaryOperation{Mul
 
         if isnothing(new_arg)
             new_arg = BinaryOperation{Mult}(args[1], args[2])
+            if can_contract(args[1], args[2])
+                new_arg = evaluate(new_arg)
+            end
         else
             new_arg = BinaryOperation{Mult}(
                 new_arg,

test/StdTest.jl

@@ -255,7 +255,7 @@ end
     @test to_std(mul(mul(mul(x, y), z), v)) == "x ⊙ y ⊙ z ⊙ v"
     @test to_std(mul(v, mul(mul(x, y), z))) == "x ⊙ y ⊙ z ⊙ v"
     @test to_std(mul(v, mul(z, mul(x, y)))) == "x ⊙ y ⊙ z ⊙ v"
-    @test to_std(mul(mul(z, v), mul(x, y))) == "z ⊙ v ⊙ x ⊙ y"
+    @test to_std(mul(mul(z, v), mul(x, y))) == "z ⊙ x ⊙ v ⊙ y" # The order changes in 'evaluate'
 end
 
 @testset "to_std output is correct with vector and trace of identity" begin