bindings: Example of using pybind11 with numpy

1 files changed, 16 insertions, 0 deletions

diff --git a/bindings/test_galearn_pdm.py b/bindings/test_galearn_pdm.py
new file mode 100644
index 0000000..43cb58e
--- /dev/null
+++ b/bindings/test_galearn_pdm.py
@@ -0,0 +1,16 @@
+
+import galearn_pdm
+import numpy
+
+def test_one():
+    
+    inp = numpy.ones(shape=10, dtype=numpy.uint8)
+    out = numpy.zeros(shape=10, dtype=numpy.int16)    
+
+    galearn_pdm.process(inp, out)
+    print(out)
+    assert out[0] == 2
+    assert out[-1] == 2
+
+if __name__ == '__main__':
+    test_one()