from __future__ import division, absolute_import, print_function

from numpy.testing import TestCase, run_module_suite, assert_equal
from pyrex_ext.primes import primes

class TestPrimes(TestCase):
    def test_simple(self, level=1):
        l = primes(10)
        assert_equal(l, [2, 3, 5, 7, 11, 13, 17, 19, 23, 29])


if __name__ == "__main__":
    run_module_suite()
