2 * Copyright (c) 2023 Samsung Electronics Co., Ltd. All Rights Reserved
3 * Copyright 2018 The TensorFlow Authors. All Rights Reserved.
5 * Licensed under the Apache License, Version 2.0 (the "License");
6 * you may not use this file except in compliance with the License.
7 * You may obtain a copy of the License at
9 * http://www.apache.org/licenses/LICENSE-2.0
11 * Unless required by applicable law or agreed to in writing, software
12 * distributed under the License is distributed on an "AS IS" BASIS,
13 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
14 * See the License for the specific language governing permissions and
15 * limitations under the License.
18 #include "kernels/Log.h"
19 #include "kernels/TestUtils.h"
20 #include "luci_interpreter/TestMemoryManager.h"
22 namespace luci_interpreter
29 using namespace testing;
31 class LogTest : public ::testing::Test
34 void SetUp() override { _memory_manager = std::make_unique<TestMemoryManager>(); }
36 std::unique_ptr<IMemoryManager> _memory_manager;
39 TEST_F(LogTest, FloatSimple)
41 std::vector<float> input_data{1, 3.1415926, 1, 1};
43 std::vector<float> ref_output_data{0, 1.14473, 0, 0};
46 makeInputTensor<DataType::FLOAT32>({1, 1, 4, 1}, input_data, _memory_manager.get());
47 Tensor output_tensor = makeOutputTensor(DataType::FLOAT32);
49 Log kernel(&input_tensor, &output_tensor);
51 _memory_manager->allocate_memory(output_tensor);
54 EXPECT_THAT(extractTensorData<float>(output_tensor), FloatArrayNear(ref_output_data));
55 EXPECT_THAT(extractTensorShape(output_tensor), ::testing::ElementsAreArray({1, 1, 4, 1}));
58 TEST_F(LogTest, Invalid_Input_Type_NEG)
60 Tensor input_tensor = makeInputTensor<DataType::S64>({1}, {1}, _memory_manager.get());
61 Tensor output_tensor = makeOutputTensor(DataType::S64);
63 Log kernel(&input_tensor, &output_tensor);
65 _memory_manager->allocate_memory(output_tensor);
66 EXPECT_ANY_THROW(kernel.execute());
70 } // namespace kernels
71 } // namespace luci_interpreter