2 from timeit
import default_timer
as clock
3 from sympy
import Symbol
, Add
, Mul
, Integer
, multinomial_coefficients
25 a
= multinomial_coefficients(4, N
)
26 b
= multinomial_coefficients(4, 2*N
)
27 t_mul
= clock() - t_mul
29 print "# of terms:", len(a2
.args
)
30 print "time doing multinomial_coefficients:", t_mul
31 print "total time2:", t_tot
33 from sympy
import ADD
, MUL
, POW
, INTEGER
, SYMBOL
38 return sympy
.Add(*[csympy2sympy(x
) for x
in a
.args
])
40 return sympy
.Mul(*[csympy2sympy(x
) for x
in a
.args
])
42 return sympy
.Pow(*[csympy2sympy(x
) for x
in a
.args
])
43 elif a
.type == INTEGER
:
44 return sympy
.sympify(str(a
))
45 elif a
.type == SYMBOL
:
46 return sympy
.sympify(str(a
))
48 raise NotImplementedError("sorry")
50 #e_sympy = csympy2sympy(e)
51 #a2_sympy = doit2(e_sympy)[0]
52 #a2_csympy = csympy2sympy(a2)
53 #print "comparison with sympy:"
54 #print a2_sympy == a2_csympy