From 46278eb0a94b040166bc2dfaf777872d42d3a703 Mon Sep 17 00:00:00 2001
From: --- <--->
Date: Sun, 27 Dec 2020 09:46:37 +0100
Subject: [PATCH 1/2] added numpy() method

---
 python/src/main.cpp | 15 +++++++++++++++
 1 file changed, 15 insertions(+)

diff --git a/python/src/main.cpp b/python/src/main.cpp
index 6f343166..871354a3 100644
--- a/python/src/main.cpp
+++ b/python/src/main.cpp
@@ -1,5 +1,6 @@
 #include <pybind11/pybind11.h>
 #include <pybind11/stl.h>
+#include <pybind11/numpy.h>
 
 #include <kompute/Kompute.hpp>
 
@@ -39,6 +40,20 @@ PYBIND11_MODULE(kp, m) {
                 return std::unique_ptr<kp::Tensor>(new kp::Tensor(data, tensorTypes));
             }), "Initialiser with list of data components and tensor GPU memory type.")
         .def("data", &kp::Tensor::data, DOC(kp, Tensor, data))
+        .def("numpy", [](kp::Tensor& self){
+                ssize_t              ndim    = 1;
+                std::vector<ssize_t> shape   = { self.size() };
+                std::vector<ssize_t> strides = { sizeof(float) };
+
+                return py::array(py::buffer_info(
+                    self.data().data(),
+                    sizeof(float),
+                    py::format_descriptor<float>::format(),
+                    ndim,
+                    shape,
+                    strides
+                ));
+            }, "Returns stored data as a new numpy array.")
         .def("__getitem__", [](kp::Tensor &self, size_t index) -> float { return self.data()[index]; },
                 "When only an index is necessary")
         .def("__setitem__", [](kp::Tensor &self, size_t index, float value) {

From 695fb08c8016633cb0fee71b5f90d99f5200a239 Mon Sep 17 00:00:00 2001
From: --- <--->
Date: Sun, 27 Dec 2020 13:26:41 +0100
Subject: [PATCH 2/2] added tests

---
 python/test/requirements-dev.txt         | 1 +
 python/test/test_array_multiplication.py | 2 ++
 python/test/test_kompute.py              | 4 ++++
 3 files changed, 7 insertions(+)

diff --git a/python/test/requirements-dev.txt b/python/test/requirements-dev.txt
index 5718a021..c28827d1 100644
--- a/python/test/requirements-dev.txt
+++ b/python/test/requirements-dev.txt
@@ -1 +1,2 @@
 pyshader==0.7.0
+numpy
diff --git a/python/test/test_array_multiplication.py b/python/test/test_array_multiplication.py
index 3ef3c02c..337c7a5d 100644
--- a/python/test/test_array_multiplication.py
+++ b/python/test/test_array_multiplication.py
@@ -1,5 +1,6 @@
 import pyshader as ps
 import kp
+import numpy as np
 
 
 def test_array_multiplication():
@@ -33,3 +34,4 @@ def compute_shader_multiply(index=("input", "GlobalInvocationId", ps.ivec3),
     mgr.eval_tensor_sync_local_def([tensor_out])
 
     assert tensor_out.data() == [2.0, 4.0, 6.0]
+    assert np.all(tensor_out.numpy() == [2.0, 4.0, 6.0])
diff --git a/python/test/test_kompute.py b/python/test/test_kompute.py
index b9b145f2..597a9d8a 100644
--- a/python/test/test_kompute.py
+++ b/python/test/test_kompute.py
@@ -1,6 +1,7 @@
 import os
 
 import kp
+import numpy as np
 
 DIRNAME = os.path.dirname(os.path.abspath(__file__))
 
@@ -22,6 +23,7 @@ def test_opmult():
     mgr.eval_tensor_sync_local_def([tensor_out])
 
     assert tensor_out.data() == [2.0, 4.0, 6.0]
+    assert np.all(tensor_out.numpy() == [2.0, 4.0, 6.0])
 
 def test_opalgobase_data():
     """
@@ -57,6 +59,7 @@ def test_opalgobase_data():
     mgr.eval_tensor_sync_local_def([tensor_out])
 
     assert tensor_out.data() == [2.0, 4.0, 6.0]
+    assert np.all(tensor_out.numpy() == [2.0, 4.0, 6.0])
 
 
 def test_opalgobase_file():
@@ -106,3 +109,4 @@ def test_sequence():
     seq.eval()
 
     assert tensor_out.data() == [2.0, 4.0, 6.0]
+    assert np.all(tensor_out.numpy() == [2.0, 4.0, 6.0])