Skip to content
Snippets Groups Projects

Bind compute dims

Merged Cyril Moineau requested to merge BindComputeDims into main
2 files
+ 17
2
Compare changes
  • Side-by-side
  • Inline
Files
2
@@ -61,5 +61,15 @@ class test_operator_binding(unittest.TestCase):
self.generic_operator.add_parameter("l_str", ["ok"])
self.assertEqual(self.generic_operator.get_parameter("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__':
unittest.main()
\ No newline at end of file
unittest.main()
Loading