Ensure doctest works when built-in function is missing.

lapets · Jun 13, 2022 · f1fd489 · f1fd489
1 parent eb1d569
commit f1fd489
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/isqrt/isqrt.py b/isqrt/isqrt.py
@@ -45,7 +45,8 @@ def isqrt(n: int) -> int:
 
     Test scenarios in which the :obj:`math.isqrt` function is not available.
 
-    >>> delattr(math, 'isqrt')
+    >>> if hasattr(math, 'isqrt'):
+    ...     delattr(math, 'isqrt')
     >>> isqrt(16)
     4
     >>> isqrt(2**30000) == 2**15000