Import cse in top module

Author: isuruf

symengine/__init__.py

@@ -11,7 +11,7 @@
         LessThan, StrictGreaterThan, StrictLessThan, Eq, Ne, Ge, Le,
         Gt, Lt, golden_ratio as GoldenRatio, catalan as Catalan,
         eulergamma as EulerGamma, Dummy, perfect_power, integer_nthroot,
-        isprime, sqrt_mod, Expr)
+        isprime, sqrt_mod, Expr, cse)
 from .utilities import var, symbols
 from .functions import *
 

symengine/lib/pywrapper.cpp

@@ -261,7 +261,7 @@ hash_t PyFunction::__hash__() const {
 bool PyFunction::__eq__(const Basic &o) const {
     if (is_a<PyFunction>(o) and
         pyfunction_class_->__eq__(*static_cast<const PyFunction &>(o).get_pyfunction_class()) and
-        unified_eq(arg_, static_cast<const PyFunction &>(o).arg_))
+        unified_eq(get_vec(), static_cast<const PyFunction &>(o).get_vec()))
         return true;
     return false;
 }
@@ -271,7 +271,7 @@ int PyFunction::compare(const Basic &o) const {
     const PyFunction &s = static_cast<const PyFunction &>(o);
     int cmp = pyfunction_class_->compare(*s.get_pyfunction_class());
     if (cmp != 0) return cmp;
-    return unified_compare(arg_, s.arg_);
+    return unified_compare(get_vec(), s.get_vec());
 }
 
 } // SymEngine