From 5686ecb8aaaf36480dc15c9af8d7ee9485b8667c Mon Sep 17 00:00:00 2001
From: cmoineau <cyril.moineau@cea.fr>
Date: Thu, 25 Jan 2024 13:02:33 +0000
Subject: [PATCH] Add unittest for magic method setter and attribute set by
kwargs.
---
aidge_core/unit_tests/test_operator_binding.py | 17 +++++++++++++++++
1 file changed, 17 insertions(+)
diff --git a/aidge_core/unit_tests/test_operator_binding.py b/aidge_core/unit_tests/test_operator_binding.py
index 825ca6100..fb7ed0587 100644
--- a/aidge_core/unit_tests/test_operator_binding.py
+++ b/aidge_core/unit_tests/test_operator_binding.py
@@ -125,6 +125,23 @@ class test_operator_binding(unittest.TestCase):
generic_op.forward() # Increment idx
self.assertEqual(customImpl.idx, 1)
+ def test_magic_meth(self):
+ myVar = 2
+ myBool = True
+ # Test dynamic attribute set
+ gop = aidge_core.GenericOperator("test", 1, 0, 1, "FictiveName", myVar=myVar).get_operator()
+ gop.myBool = myBool
+ # Test variable set by kwargs
+ self.assertEqual(gop.myVar, myVar)
+ # Test set attr
+ self.assertEqual(gop.myBool, myBool)
+
+ # Test static attribute set !
+ prod = aidge_core.Producer([1]).get_operator()
+ self.assertEqual(prod.Constant, False)
+ prod.Constant = True # By default Constant is False
+ self.assertEqual(prod.Constant, True)
+
if __name__ == '__main__':
--
GitLab