1 /** 2 * Copyright 2020 Huawei Technologies Co., Ltd 3 * 4 * Licensed under the Apache License, Version 2.0 (the "License"); 5 * you may not use this file except in compliance with the License. 6 * You may obtain a copy of the License at 7 * 8 * http://www.apache.org/licenses/LICENSE-2.0 9 * 10 * Unless required by applicable law or agreed to in writing, software 11 * distributed under the License is distributed on an "AS IS" BASIS, 12 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. 13 * See the License for the specific language governing permissions and 14 * limitations under the License. 15 */ 16 #include "common/common.h" 17 #include "gtest/gtest.h" 18 #include "securec.h" 19 #include "minddata/dataset/engine/perf/cyclic_array.h" 20 #include "minddata/dataset/engine/perf/perf_data.h" 21 22 using namespace mindspore::dataset; 23 24 class MindDataTestPerfData : public UT::Common { 25 public: 26 MindDataTestPerfData() {} 27 }; 28 29 TEST_F(MindDataTestPerfData, Test1) { 30 PerfData<std::vector<int64_t>> p1(2, 3); 31 PerfData<CyclicArray<int>> p2(2, 3); 32 EXPECT_EQ(p1.capacity(), p2.capacity()); 33 std::vector<int64_t> row = {1, 2, 3}; 34 p1.AddSample(row); 35 p2.AddSample(row); 36 EXPECT_EQ(p1.size(), 1); 37 EXPECT_EQ(p1.size(), p2.size()); 38 p1.AddSample(row); 39 p2.AddSample(row); 40 EXPECT_EQ(p1.size(), 2); 41 EXPECT_EQ(p1.size(), p2.size()); 42 std::vector<int64_t> row1 = {4, 5, 6}; 43 p2.AddSample(row1); 44 EXPECT_EQ(p2.size(), 2); 45 auto r1 = p2.Row<int>(static_cast<int64_t>(0)); 46 for (auto i = 0; i < 3; i++) { 47 EXPECT_EQ(r1[i], i + 1); 48 } 49 50 auto r2 = p2.Row<int>(1); 51 for (auto i = 0; i < 3; i++) { 52 EXPECT_EQ(r2[i], i + 4); 53 } 54 55 EXPECT_EQ(p2[0][1], 4); 56 EXPECT_EQ(p2[1][1], 5); 57 EXPECT_EQ(p2[2][1], 6); 58 } 59 60 TEST_F(MindDataTestPerfData, Test2) { 61 auto pd = PerfData<CyclicArray<int>>(1000000, 3); 62 auto row = {1, 2, 3}; 63 pd.AddSample(row); 64 EXPECT_EQ(pd[0][0], 1); 65 EXPECT_EQ(pd[1][0], 2); 66 EXPECT_EQ(pd[2][0], 3); 67 auto row1 = {4, 5, 6}; 68 pd.AddSample(row1); 69 EXPECT_EQ(pd[0][0], 1); 70 EXPECT_EQ(pd[1][0], 2); 71 EXPECT_EQ(pd[2][0], 3); 72 EXPECT_EQ(pd[0][1], 4); 73 EXPECT_EQ(pd[1][1], 5); 74 EXPECT_EQ(pd[2][1], 6); 75 }