Commit 4840aa33 authored by 余昆's avatar 余昆
Browse files

fix test_ivf bug

parent e728c699

core/src/index/unittest/test_ivf.cpp

+5 −3
Original line number Diff line number Diff line
@@ -181,11 +181,13 @@ TEST_P(IVFTest, clone_test) { 
    // PrintResult(result, nq, k);



    auto AssertEqual = [&](knowhere::DatasetPtr p1, knowhere::DatasetPtr p2) {

        auto ids_p1 = p1->array()[0];

        auto ids_p2 = p2->array()[0];

        auto ids_p1 = p1->ids();

        auto ids_p2 = p2->ids();



        for (int i = 0; i < nq * k; ++i) {

            EXPECT_EQ(*(ids_p2->data()->GetValues<int64_t>(1, i)), *(ids_p1->data()->GetValues<int64_t>(1, i)));

            EXPECT_EQ(*((int64_t*)(ids_p2) + i), *((int64_t*)(ids_p1) + i));

            //            EXPECT_EQ(*(ids_p2->data()->GetValues<int64_t>(1, i)), *(ids_p1->data()->GetValues<int64_t>(1,

            //            i)));

        }

    };