Add an example of the same wrapper using cython (mathbunnyru#2)

* Add an example of the same wrapper using cython

* Update .gitignore

* Update cython_sieve.pyx

* Update sieve.pxd

Co-authored-by: Ayaz Salikhov <[email protected]>

Author: moskupols

.gitignore

@@ -130,3 +130,6 @@ dmypy.json
 
 # macOS files
 .DS_Store
+
+# Cython generated file
+cython_sieve.cpp

cython_sieve.pyx

@@ -0,0 +1,8 @@
+cimport sieve
+
+
+def sieve_of_eratosthenes(size_t n):
+    '''
+    Return list of prime numbers until given number. Implemented in Cython.
+    '''
+    return sieve.SieveOfEratosthenes(n)

setup.py

@@ -1,6 +1,8 @@
 from setuptools import setup, Extension
 import sysconfig
 
+from Cython.Build import cythonize
+
 
 language = 'c++'
 std = 'c++17'
@@ -18,9 +20,16 @@
     language=language
 )
 
+cython_extension = Extension(
+    'cython_sieve',
+    sources=['cython_sieve.pyx', 'sieve.cpp'],
+    extra_compile_args=extra_compile_args,
+    language='c++',
+)
+
 setup(
     name='cpp_python_extension',
     version='1.0',
     description='This is Example module written in C++',
-    ext_modules=[extension]
+    ext_modules=cythonize([extension, cython_extension]),
 )

sieve.pxd

@@ -0,0 +1,5 @@
+from libcpp.vector cimport vector
+
+
+cdef extern from "sieve.h" namespace "abc":
+    vector[size_t] SieveOfEratosthenes(size_t n)

test_benchmark.py

@@ -1,7 +1,12 @@
 import pytest
 import cpp_python_extension
+import cython_sieve
 
 
 @pytest.mark.parametrize("input", [1000, 100000, 1000000])
 def test_cpp(benchmark, input):
     benchmark(cpp_python_extension.sieve_of_eratosthenes, input)
+
+@pytest.mark.parametrize("input", [1000, 100000, 1000000])
+def test_cyth(benchmark, input):
+    benchmark(cython_sieve.sieve_of_eratosthenes, input)