Add lambdify support for harmonic()

smichr · Aug 30, 2022 · e63a2e6 · e63a2e6
1 parent aeaeea6
commit e63a2e6
Show file tree

Hide file tree

Showing 2 changed files with 14 additions and 1 deletion.
diff --git a/sympy/printing/numpy.py b/sympy/printing/numpy.py
@@ -404,6 +404,9 @@ def _print_bernoulli(self, expr):
         # scipy's bernoulli is inconsistent with SymPy's so rewrite
         return self._print(expr._eval_rewrite_as_zeta(*expr.args))
 
+    def _print_harmonic(self, expr):
+        return self._print(expr._eval_rewrite_as_zeta(*expr.args))
+
     def _print_Integral(self, e):
         integration_vars, limits = _unpack_integral_limits(e)
 

diff --git a/sympy/utilities/tests/test_lambdify.py b/sympy/utilities/tests/test_lambdify.py
@@ -11,7 +11,7 @@
 from sympy.core.singleton import S
 from sympy.core.symbol import (Dummy, symbols)
 from sympy.functions.combinatorial.factorials import (RisingFactorial, factorial)
-from sympy.functions.combinatorial.numbers import bernoulli
+from sympy.functions.combinatorial.numbers import bernoulli, harmonic
 from sympy.functions.elementary.complexes import Abs
 from sympy.functions.elementary.exponential import exp, log
 from sympy.functions.elementary.hyperbolic import acosh
@@ -1476,6 +1476,16 @@ def test_scipy_bernoulli():
     assert bern(1) == 0.5
 
 
+def test_scipy_harmonic():
+    if not scipy:
+        skip("scipy not installed")
+
+    hn = lambdify((x,), harmonic(x), modules='scipy')
+    assert hn(2) == 1.5
+    hnm = lambdify((x, y), harmonic(x, y), modules='scipy')
+    assert hnm(2, 2) == 1.25
+
+
 def test_cupy_array_arg():
     if not cupy:
         skip("CuPy not installed")