Skip to content
Snippets Groups Projects

Unified interface for attributes

Merged Olivier BICHLER requested to merge unified_params into main
All threads resolved!
4 files
+ 30
12
Compare changes
  • Side-by-side
  • Inline
Files
4
@@ -67,5 +67,15 @@ class test_operator_binding(unittest.TestCase):
@@ -67,5 +67,15 @@ class test_operator_binding(unittest.TestCase):
self.generic_operator.add_attr("l_str", ["ok"])
self.generic_operator.add_attr("l_str", ["ok"])
self.assertEqual(self.generic_operator.get_attr("l_str"), ["ok"])
self.assertEqual(self.generic_operator.get_attr("l_str"), ["ok"])
 
def test_compute_output_dims(self):
 
in_dims=[25, 25]
 
input = aidge_core.Producer(in_dims, name="In")
 
genOp = aidge_core.GenericOperator("genOp", 1, 1, 1, name="genOp")
 
_ = aidge_core.sequential([input, genOp])
 
self.assertListEqual(genOp.get_operator().output(0).dims(), [])
 
genOp.get_operator().set_compute_output_dims(lambda x:x)
 
genOp.get_operator().compute_output_dims()
 
self.assertListEqual(genOp.get_operator().output(0).dims(), in_dims)
 
if __name__ == '__main__':
if __name__ == '__main__':
unittest.main()
unittest.main()
\ No newline at end of file
Loading