Return NotImplemented when non-default mod is passed to __pow__ and __rpow__

Author: antonwolfy

dpnp/dpnp_array.py

@@ -480,6 +480,8 @@ def __pos__(self, /):
 
     def __pow__(self, other, mod=None, /):
         r"""Return :math:`\text{self ** value}`."""
+        if mod is not None:
+            return NotImplemented
         return dpnp.power(self, other)
 
     def __radd__(self, other, /):
@@ -524,6 +526,8 @@ def __ror__(self, other, /):
 
     def __rpow__(self, other, mod=None, /):
         r"""Return :math:`\text{value ** self}`."""
+        if mod is not None:
+            return NotImplemented
         return dpnp.power(other, self)
 
     def __rrshift__(self, other, /):

dpnp/tests/test_ndarray.py

@@ -550,3 +550,15 @@ def test_rmatmul_numpy_array():
 
     with pytest.raises(TypeError):
         b @ a
+
+
+@pytest.mark.parametrize("xp", [dpnp, numpy])
+def test_pow_modulo(xp):
+    a = xp.array([2, 3, 4])
+    b = xp.array([5, 2, 3])
+
+    assert a.__pow__(b, 10) == NotImplemented
+    assert a.__rpow__(b, 10) == NotImplemented
+
+    assert (a.__pow__(b, None) == a**b).all()
+    assert (a.__rpow__(b, None) == b**a).all()