@@ -291,7 +291,7 @@ TEST_CASE("TestBinnedExpressions")
291291 auto pspecs = createOperations (p);
292292 auto schema = std::make_shared<arrow::Schema>(std::vector{o2::aod::track::Pt::asArrowField (), o2::aod::track::X::asArrowField (), o2::aod::track::Y::asArrowField (), o2::aod::track::Z::asArrowField (), o2::aod::track::Phi::asArrowField ()});
293293 auto tree = createExpressionTree (pspecs, schema);
294- REQUIRE(tree->ToString() == "if (bool less_than((float) fPt, (const float) 0.5 raw(3f000000))) { (const float) 0 raw(0) } else { if (bool greater_than((float) fPt, (const float) 0.5 raw(3f000000)) && bool less_than_or_equal_to((float) fPt, (const float) 1.5 raw(3fc00000))) { float add(float add(float add(float multiply((const float) 1 raw(3f800000), (float) fX), float multiply((const float) 1.1 raw(3f8ccccd), (float) fY)), float multiply((const float) 1.2 raw(3f99999a), (float) fZ)), float multiply((const float) 1.3 raw(3fa66666), (float) fPhi)) } else { if (bool less_than_or_equal_to((float) fPt, (const float) 2.5 raw(40200000))) { float add(float add(float add(float multiply((const float) 2 raw(40000000), (float) fX), float multiply((const float) 2.1 raw(40066666), (float) fY)), float multiply((const float) 2.2 raw(400ccccd), (float) fZ)), float multiply((const float) 2.3 raw(40133333), (float) fPhi)) } else { if (bool less_than_or_equal_to((float) fPt, (const float) 3.5 raw(40600000))) { float add(float add(float add(float multiply((const float) 3 raw(40400000), (float) fX), float multiply((const float) 3.1 raw(40466666), (float) fY)), float multiply((const float) 3.2 raw(404ccccd), (float) fZ)), float multiply((const float) 3.3 raw(40533333), (float) fPhi)) } else { if (bool less_than_or_equal_to((float) fPt, (const float) 4.5 raw(40900000))) { float add(float add(float add(float multiply((const float) 4 raw(40800000), (float) fX), float multiply((const float) 4.1 raw(40833333), (float) fY)), float multiply((const float) 4.2 raw(40866666), (float) fZ)), float multiply((const float) 4.3 raw(4089999a), (float) fPhi)) } else { (const float) 0 raw(0) } } } } }");
294+ REQUIRE(tree->ToString() == "if (bool less_than((float) fPt, (const float) 0.5 raw(3f000000))) { (const float) 0 raw(0) } else { if (bool less_than((float) fPt, (const float) 1.5 raw(3fc00000))) { float add(float add(float add(float multiply((const float) 1 raw(3f800000), (float) fX), float multiply((const float) 1.1 raw(3f8ccccd), (float) fY)), float multiply((const float) 1.2 raw(3f99999a), (float) fZ)), float multiply((const float) 1.3 raw(3fa66666), (float) fPhi)) } else { if (bool less_than((float) fPt, (const float) 2.5 raw(40200000))) { float add(float add(float add(float multiply((const float) 2 raw(40000000), (float) fX), float multiply((const float) 2.1 raw(40066666), (float) fY)), float multiply((const float) 2.2 raw(400ccccd), (float) fZ)), float multiply((const float) 2.3 raw(40133333), (float) fPhi)) } else { if (bool less_than((float) fPt, (const float) 3.5 raw(40600000))) { float add(float add(float add(float multiply((const float) 3 raw(40400000), (float) fX), float multiply((const float) 3.1 raw(40466666), (float) fY)), float multiply((const float) 3.2 raw(404ccccd), (float) fZ)), float multiply((const float) 3.3 raw(40533333), (float) fPhi)) } else { if (bool less_than((float) fPt, (const float) 4.5 raw(40900000))) { float add(float add(float add(float multiply((const float) 4 raw(40800000), (float) fX), float multiply((const float) 4.1 raw(40833333), (float) fY)), float multiply((const float) 4.2 raw(40866666), (float) fZ)), float multiply((const float) 4.3 raw(4089999a), (float) fPhi)) } else { (const float) 0 raw(0) } } } } }");
295295
296296 std::vector<float > binning{0 , o2::constants::math::PIHalf, o2::constants::math::PI, o2::constants::math::PI + o2::constants::math::PIHalf, o2::constants::math::TwoPI};
297297 std::vector<float > parameters{1.0 , 1.1 , 1.2 , 1.3 , // par 0
@@ -306,5 +306,5 @@ TEST_CASE("TestBinnedExpressions")
306306 auto p2specs = createOperations (p2);
307307 auto schema2 = std::make_shared<arrow::Schema>(std::vector{o2::aod::track::Phi::asArrowField (), o2::aod::track::X::asArrowField (), o2::aod::track::Y::asArrowField (), o2::aod::track::Z::asArrowField ()});
308308 auto tree2 = createExpressionTree (p2specs, schema2);
309- REQUIRE(tree2->ToString() == "if (bool less_than((float) fPhi, (const float) 0 raw(0))) { (const float) -1 raw(bf800000) } else { if (bool greater_than((float) fPhi, (const float) 0 raw(0)) && bool less_than_or_equal_to((float) fPhi, (const float) 1.5708 raw(3fc90fdb))) { float add(float add(float multiply(float multiply((const float) 1 raw(3f800000), (float) fX), (float) fX), float multiply(float multiply((const float) 1.1 raw(3f8ccccd), (float) fY), (float) fY)), float multiply(float multiply((const float) 1.2 raw(3f99999a), (float) fZ), (float) fZ)) } else { if (bool less_than_or_equal_to((float) fPhi, (const float) 3.14159 raw(40490fdb))) { float add(float add(float multiply(float multiply((const float) 2 raw(40000000), (float) fX), (float) fX), float multiply(float multiply((const float) 2.1 raw(40066666), (float) fY), (float) fY)), float multiply(float multiply((const float) 2.2 raw(400ccccd), (float) fZ), (float) fZ)) } else { if (bool less_than_or_equal_to((float) fPhi, (const float) 4.71239 raw(4096cbe4))) { float add(float add(float multiply(float multiply((const float) 3 raw(40400000), (float) fX), (float) fX), float multiply(float multiply((const float) 3.1 raw(40466666), (float) fY), (float) fY)), float multiply(float multiply((const float) 3.2 raw(404ccccd), (float) fZ), (float) fZ)) } else { if (bool less_than_or_equal_to((float) fPhi, (const float) 6.28319 raw(40c90fdb))) { float add(float add(float multiply(float multiply((const float) 4 raw(40800000), (float) fX), (float) fX), float multiply(float multiply((const float) 4.1 raw(40833333), (float) fY), (float) fY)), float multiply(float multiply((const float) 4.2 raw(40866666), (float) fZ), (float) fZ)) } else { (const float) -1 raw(bf800000) } } } } }");
309+ REQUIRE(tree2->ToString() == "if (bool less_than((float) fPhi, (const float) 0 raw(0))) { (const float) -1 raw(bf800000) } else { if (bool less_than((float) fPhi, (const float) 1.5708 raw(3fc90fdb))) { float add(float add(float multiply(float multiply((const float) 1 raw(3f800000), (float) fX), (float) fX), float multiply(float multiply((const float) 1.1 raw(3f8ccccd), (float) fY), (float) fY)), float multiply(float multiply((const float) 1.2 raw(3f99999a), (float) fZ), (float) fZ)) } else { if (bool less_than((float) fPhi, (const float) 3.14159 raw(40490fdb))) { float add(float add(float multiply(float multiply((const float) 2 raw(40000000), (float) fX), (float) fX), float multiply(float multiply((const float) 2.1 raw(40066666), (float) fY), (float) fY)), float multiply(float multiply((const float) 2.2 raw(400ccccd), (float) fZ), (float) fZ)) } else { if (bool less_than((float) fPhi, (const float) 4.71239 raw(4096cbe4))) { float add(float add(float multiply(float multiply((const float) 3 raw(40400000), (float) fX), (float) fX), float multiply(float multiply((const float) 3.1 raw(40466666), (float) fY), (float) fY)), float multiply(float multiply((const float) 3.2 raw(404ccccd), (float) fZ), (float) fZ)) } else { if (bool less_than((float) fPhi, (const float) 6.28319 raw(40c90fdb))) { float add(float add(float multiply(float multiply((const float) 4 raw(40800000), (float) fX), (float) fX), float multiply(float multiply((const float) 4.1 raw(40833333), (float) fY), (float) fY)), float multiply(float multiply((const float) 4.2 raw(40866666), (float) fZ), (float) fZ)) } else { (const float) -1 raw(bf800000) } } } } }");
310310}
0 commit comments