add name property to FunctionSymbol

symengine · Jan 23, 2024 · a165060 · a165060
1 parent a7136d7
commit a165060
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 1 deletion.
diff --git a/symengine/lib/symengine_wrapper.in.pyx b/symengine/lib/symengine_wrapper.in.pyx
@@ -2841,6 +2841,10 @@ class FunctionSymbol(Function):
         name = deref(X).get_name().decode("utf-8")
         return str(name)
 
+    @property
+    def name(Basic self):
+        return self.get_name()
+
     def _sympy_(self):
         import sympy
         name = self.get_name()

diff --git a/symengine/tests/test_functions.py b/symengine/tests/test_functions.py
@@ -105,12 +105,15 @@ def test_derivative():
 
 def test_function():
     x = Symbol("x")
-    assert Function("f")(x) == function_symbol("f", x)
+    fx = Function("f")(x)
+    assert fx == function_symbol("f", x)
 
     raises(TypeError, lambda: Function("f", "x"))
     raises(TypeError, lambda: Function("f", x))
     raises(TypeError, lambda: Function())
 
+    assert fx.name == "f"
+
 
 def test_abs():
     x = Symbol("x")