From fc5a2f7f6d8061ddcfc890c0ac56613cf7e9bdc5 Mon Sep 17 00:00:00 2001
From: cmoineau <cyril.moineau@cea.fr>
Date: Wed, 20 Sep 2023 06:26:35 +0000
Subject: [PATCH] [TensorImpl] Add unit test on get/set methods.
---
aidge_backend_cpu/unit_tests/test_tensor.py | 12 ++++++++++++
1 file changed, 12 insertions(+)
diff --git a/aidge_backend_cpu/unit_tests/test_tensor.py b/aidge_backend_cpu/unit_tests/test_tensor.py
index 1d12fc0c..438b6acd 100644
--- a/aidge_backend_cpu/unit_tests/test_tensor.py
+++ b/aidge_backend_cpu/unit_tests/test_tensor.py
@@ -45,5 +45,17 @@ class test_tensor(unittest.TestCase):
self.assertTrue(i_t == i_n) # TODO : May need to change this to a difference
for i,j in zip(t.dims(), np_array.shape):
self.assertEqual(i,j)
+
+ def test_get_set(self):
+ dims = [2,2,2]
+
+ np_array = np.arange(8).reshape(dims)
+ # Numpy -> Tensor
+ t = aidge_core.Tensor(np_array)
+ for i in range(8):
+ self.assertEqual(t[i], i)
+ t[i] = 5
+ self.assertEqual(t[i], 5)
+
if __name__ == '__main__':
unittest.main()
--
GitLab