diff --git a/aidge_core/unit_tests/test_operator_binding.py b/aidge_core/unit_tests/test_operator_binding.py index 825ca6100382116443699a00bcff27b9bbca028a..fb7ed0587fb074858e9f3766d5de0d43b39d1ef5 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__':