Add support for numpy array attributes (#7)

Add support for numpy array attributes

Author: mathias-nillion

nada_algebra/array.py

@@ -32,10 +32,23 @@ class NadaArray:
     inner: np.ndarray
 
     SUPPORTED_OPERATIONS = {
+        "base",
         "compress",
         "copy",
         "cumprod",
         "cumsum",
+        "data",
+        "dtype",
+        "diagonal",
+        "fill",
+        "flags",
+        "flat",
+        "flatten",
+        "item",
+        "itemset",
+        "itemsize",
+        "nbytes",
+        "ndim",
         "diagonal",
         "fill",
         "flatten",
@@ -49,6 +62,7 @@ class NadaArray:
         "resize",
         "shape",
         "size",
+        "strides",
         "squeeze",
         "sum",
         "swapaxes",
@@ -82,7 +96,6 @@ def __setitem__(self, key, value):
             value: The value to set.
         """
         if isinstance(value, NadaArray):
-            # print("NadaArray")
             self.inner[key] = value.inner
         else:
             self.inner[key] = value

tests/nada-tests/src/supported_operations.py

@@ -7,6 +7,15 @@ def nada_main():
 
     a = na.array([3, 3], parties[0], "A")
 
+    assert isinstance(a.data, memoryview)
+    assert a.dtype == NadaType
+    assert a.flags["WRITEABLE"]
+    assert isinstance(na.NadaArray(a.flat), na.NadaArray)
+    assert a.itemsize == 8
+    assert a.nbytes == 72
+    assert a.ndim == 2
+    assert a.strides == (24, 8)
+
     try:
         a.argsort()
         raise Exception("Unsopported operation `argsort` occurred")

tests/test_all.py

@@ -15,6 +15,8 @@
     "vstack",
     "reveal",
     "matrix_multiplication",
+    "generate_array",
+    "supported_operations",
     "get_item",
     "get_attr",
     "set_item",