#include "gtest/gtest.h"
#include <near_set.h>

TEST(NearSet, Time) {
    NearSet my_set;
    for (int i = 0; i < 100000; ++i) {
        my_set.Add((int64_t)i);
    }

    for (int i = 0; i < 100000; ++i) {
        int64_t x = (int64_t)i;
        std::vector<int64_t> ans = my_set.FindNear(x, 0);
        ASSERT_EQ((int)ans.size(), 1);
        ASSERT_EQ(x, ans[0]);
    }
}

TEST(NearSet, TimeFive) {
    NearSet my_set;
    for (int i = 0; i < 40000; ++i) {
        my_set.Add((int64_t)i);
    }

    for (int i = 0; i < 39999; ++i) {
        my_set.Remove((int64_t)i);
        std::vector<int64_t>ans = my_set.FindNear((int64_t)i, 1);
        ASSERT_EQ((int)ans.size(), 1);
    }
}