added test for push_children (tensor IR nodes)

1 files changed, 24 insertions, 0 deletions

diff --git a/eval/src/tests/eval/tensor_function/tensor_function_test.cpp b/eval/src/tests/eval/tensor_function/tensor_function_test.cpp
index 681a4dabc19..641ebddfec2 100644
--- a/eval/src/tests/eval/tensor_function/tensor_function_test.cpp
+++ b/eval/src/tests/eval/tensor_function/tensor_function_test.cpp
@@ -162,4 +162,28 @@ TEST("require that tensor join works") {
     TEST_DO(verify_equal(*expect, ctx.eval(prog)));
 }
 
+TEST("require that push_children works") {
+    Stash stash;
+    std::vector<Node::Child::CREF> refs;
+    const Node &a = inject(ValueType::double_type(), 0, stash);
+    const Node &b = inject(ValueType::double_type(), 1, stash);
+    a.push_children(refs);
+    b.push_children(refs);
+    ASSERT_EQUAL(refs.size(), 0u);
+    //-------------------------------------------------------------------------
+    reduce(a, Aggr::SUM, {}, stash).push_children(refs);
+    ASSERT_EQUAL(refs.size(), 1u);
+    EXPECT_EQUAL(&refs[0].get().get(), &a);
+    //-------------------------------------------------------------------------
+    map(b, operation::Neg::f, stash).push_children(refs);
+    ASSERT_EQUAL(refs.size(), 2u);
+    EXPECT_EQUAL(&refs[1].get().get(), &b);
+    //-------------------------------------------------------------------------
+    join(a, b, operation::Add::f, stash).push_children(refs);
+    ASSERT_EQUAL(refs.size(), 4u);
+    EXPECT_EQUAL(&refs[2].get().get(), &a);
+    EXPECT_EQUAL(&refs[3].get().get(), &b);
+    //-------------------------------------------------------------------------
+}
+
 TEST_MAIN() { TEST_RUN_ALL(); }