60 lines
1.6 KiB
C++
60 lines
1.6 KiB
C++
/*
|
|
* Copyright (c) Facebook, Inc. and its affiliates.
|
|
*
|
|
* Licensed under the Apache License Version 2.0 with LLVM Exceptions
|
|
* (the "License"); you may not use this file except in compliance with
|
|
* the License. You may obtain a copy of the License at
|
|
*
|
|
* https://llvm.org/LICENSE.txt
|
|
*
|
|
* Unless required by applicable law or agreed to in writing, software
|
|
* distributed under the License is distributed on an "AS IS" BASIS,
|
|
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
|
* See the License for the specific language governing permissions and
|
|
* limitations under the License.
|
|
*/
|
|
#include <unifex/reduce_stream.hpp>
|
|
|
|
#include <unifex/sync_wait.hpp>
|
|
#include <unifex/transform_stream.hpp>
|
|
#include <unifex/then.hpp>
|
|
#include <unifex/range_stream.hpp>
|
|
|
|
#include <cstdio>
|
|
|
|
#include <gtest/gtest.h>
|
|
|
|
using namespace unifex;
|
|
|
|
TEST(reduce_stream, Smoke) {
|
|
int finalResult;
|
|
|
|
sync_wait(then(
|
|
reduce_stream(
|
|
transform_stream(
|
|
range_stream{0, 10},
|
|
[](int value) { return value * value; }),
|
|
0,
|
|
[](int state, int value) { return state + value; }),
|
|
[&](int result) { finalResult = result; }));
|
|
|
|
EXPECT_EQ(finalResult, 285);
|
|
std::printf("result = %i\n", finalResult);
|
|
}
|
|
|
|
TEST(reduce_stream, Pipeable) {
|
|
int finalResult;
|
|
|
|
range_stream{0, 10}
|
|
| transform_stream(
|
|
[](int value) { return value * value; })
|
|
| reduce_stream(
|
|
0,
|
|
[](int state, int value) { return state + value; })
|
|
| then([&](int result) { finalResult = result; })
|
|
| sync_wait();
|
|
|
|
EXPECT_EQ(finalResult, 285);
|
|
std::printf("result = %i\n", finalResult);
|
|
}
|