From 70ce7ab948cf09847c5ad5dbbdfa00f0f52d2c89 Mon Sep 17 00:00:00 2001
From: neingeist <neingeist@bl0rg.net>
Date: Mon, 1 Sep 2014 23:03:51 +0200
Subject: [PATCH] play around with cython

---
 cython_test.pyx     | 16 ++++++++++++++++
 cython_test_test.py | 45 +++++++++++++++++++++++++++++++++++++++++++++
 2 files changed, 61 insertions(+)
 create mode 100644 cython_test.pyx
 create mode 100644 cython_test_test.py

diff --git a/cython_test.pyx b/cython_test.pyx
new file mode 100644
index 0000000..b924c83
--- /dev/null
+++ b/cython_test.pyx
@@ -0,0 +1,16 @@
+def say_hello_to(name):
+  print("Hello %s!" % name)
+
+from math import sin
+
+def f(double x):
+    return sin(x**2)
+
+def integrate_f(double a, double b, int N):
+    cdef int i
+    cdef double s, dx
+    s = 0
+    dx = (b-a)/N
+    for i in range(N):
+        s += f(a+i*dx)
+    return s * dx
diff --git a/cython_test_test.py b/cython_test_test.py
new file mode 100644
index 0000000..672c3a3
--- /dev/null
+++ b/cython_test_test.py
@@ -0,0 +1,45 @@
+from __future__ import division, print_function
+from math import sin
+from random import uniform, randint
+from timeit import timeit
+
+import pyximport
+pyximport.install()
+
+import cython_test
+
+
+# as defined in cython_test.pyx, but in pure python:
+def f(x):
+    return sin(x**2)
+
+
+# as defined in cython_test.pyx, but in pure python:
+def integrate_f(a, b, N):
+    s = 0
+    dx = (b-a)/N
+    for i in range(N):
+        s += f(a+i*dx)
+    return s * dx
+
+if __name__ == "__main__":
+    cython_test.say_hello_to("schweini")
+
+    assert(cython_test.f(0.5) == f(0.5))
+    assert(cython_test.f(1.7) == f(1.7))
+
+    for i in range(10):
+        a = uniform(-3.14, 3.14)
+        b = uniform(-3.14, 3.14)
+        N = randint(10, 1000)
+        assert(integrate_f(a, b, N)
+               == cython_test.integrate_f(a, b, N))
+
+    print("Python:",
+          timeit('integrate_f(0.0, 0.25, 1000)',
+                 'from cython_test_test import integrate_f',
+                 number=10000))
+    print("Cython:",
+          timeit('cython_test.integrate_f(0.0, 0.25, 1000)',
+                 'import pyximport; pyximport.install(); import cython_test',
+                 number=10000))