Feature379 ckksvector shape (#394)

* overridden property `shape` for `CKKSVector`

* added test for `shape` on `CKKSVector`

* applied black

tenseal/tensors/ckksvector.py

@@ -52,6 +52,10 @@ def decrypt(self, secret_key: "ts.enc_context.SecretKey" = None) -> List[float]:
     def size(self) -> int:
         return self.data.size()
 
+    @property
+    def shape(self) -> List[int]:
+        return [self.size()]
+
     def ciphertext(self) -> List["ts._ts_cpp.Ciphertext"]:
         return self.data.ciphertext()
 

tests/python/tenseal/tensors/test_ckks_vector.py

@@ -1282,3 +1282,9 @@ def test_size(context):
     for size in range(1, 10):
         vec = ts.ckks_vector(context, [1] * size)
         assert vec.size() == size, "Size of encrypted vector is incorrect."
+
+
+def test_shape(context):
+    for size in range(1, 10):
+        vec = ts.ckks_vector(context, [1] * size)
+        assert vec.shape == [size], "Shape of encrypted vector is incorrect."