From 6e3f78d6e0cd5805043a56569584af8eab154437 Mon Sep 17 00:00:00 2001 From: Marcel Koch Date: Thu, 15 Aug 2024 11:51:08 +0200 Subject: [PATCH] [core] add tests for lambda op name --- core/test/base/executor.cpp | 38 +++++++++++++++++++++++++++++++++++++ 1 file changed, 38 insertions(+) diff --git a/core/test/base/executor.cpp b/core/test/base/executor.cpp index 64a11929983..20f795b2ded 100644 --- a/core/test/base/executor.cpp +++ b/core/test/base/executor.cpp @@ -521,4 +521,42 @@ TEST_F(ExecutorLogging, LogsOperation) } +struct NameLogger : public gko::log::Logger { +protected: + void on_operation_launched(const gko::Executor* exec, + const gko::Operation* op) const override + { + op_name = op->get_name(); + } + +public: + mutable std::string op_name; +}; + + +TEST(LambdaOperation, CanSetName) +{ + auto name_logger = std::make_shared(); + auto exec = gko::ReferenceExecutor::create(); + exec->add_logger(name_logger); + + exec->run( + "name", [] {}, [] {}, [] {}, [] {}); + + ASSERT_EQ("name", name_logger->op_name); +} + + +TEST(LambdaOperation, HasDefaultName) +{ + auto name_logger = std::make_shared(); + auto exec = gko::ReferenceExecutor::create(); + exec->add_logger(name_logger); + + exec->run([] {}, [] {}, [] {}, [] {}); + + ASSERT_EQ("unname", name_logger->op_name); +} + + } // namespace