Added number.is_prime() method

CHANGELOG.rst

@@ -22,6 +22,7 @@ Changelog
 - Add ``is_multiple_of()``.
 - Add ``is_number()``.
 - Add ``is_odd()``.
+- Add ``is_prime()``.
 - Add ``is_string()``.
 - Add ``last()``.
 - Add ``random_()``.

docs/example.rst

@@ -166,6 +166,15 @@ Numbers
     >>> _.is_odd(7)
     True
 
+    >>> _.is_prime(5)
+    True
+    >>> _.is_prime(7)
+    True
+    >>> _.is_prime(4)
+    False
+    >>> _.is_prime(727021)
+    True
+
     >>> result = _.random_(5, 6)
     >>> assert 5 <= result <= 6
     >>> result = _.random_(5)

sugar/__init__.py

@@ -36,6 +36,7 @@
     is_even,
     is_multiple_of,
     is_odd,
+    is_prime,
     random_
 )
 

sugar/_compat.py

@@ -20,6 +20,8 @@
 if PY3:
     string_types = (str,)
     number_types = (int, float, Decimal)
+    _range = range
 else:
     string_types = (str, unicode)
     number_types = (int, long, float, Decimal)
+    _range = xrange

sugar/arrays.py

@@ -4,6 +4,8 @@
 
 import copy
 
+from ._compat import _range
+
 import sugar as _
 
 
@@ -153,7 +155,7 @@ def construct(var, callback):
 
     .. versionadded:: 0.1.0
     """
-    return [callback(i) for i in range(0, var)]
+    return [callback(i) for i in _range(0, var)]
 
 
 def count(array, value):

sugar/number.py

@@ -2,10 +2,11 @@
 
 from __future__ import absolute_import
 
-import sys
 import random
 import math
 
+from ._compat import _range
+
 
 def is_even(num):
     """Returns True if :attr:`num` is even.
@@ -75,6 +76,42 @@ def is_odd(num):
     return num % 2 != 0
 
 
+def is_prime(num):
+    """Returns True if the give :attr:`num` is a prime number.
+
+    Args:
+        num (int/float): Number passed in by the user.
+
+    Returns:
+        bool: True if the given :attr:`num` is a prime number else False
+
+    Example:
+
+        >>> is_prime(5)
+        True
+        >>> is_prime(7)
+        True
+        >>> is_prime(4)
+        False
+        >>> is_prime(727021)
+        True
+
+    .. versionadded:: TODO
+    """
+    count = 0
+
+    for i in _range(1, (math.ceil(math.sqrt(num)) + 1)):
+        if num % i == 0:
+            count += 1
+
+    if count <= 1:
+        result = True
+    else:
+        result = False
+
+    return result
+
+
 def random_(n1=None, n2=None):
     """Returns a random integer/float from :attr:`n1` to :attr:`n2` (both
     inclusive)

tests/test_number.py

@@ -30,6 +30,17 @@ def test_is_odd(num, expected):
     assert _.is_odd(num) == expected
 
 
+@parametrize('num,expected', [
+    (13, True),
+    (727021, True),
+    (10, False),
+    (21, False),
+    (37, True)
+])
+def test_is_prime(num, expected):
+    assert _.is_prime(num) == expected
+
+
 @parametrize('args', [
     (0, 5),
     (2, 5),