Spaces:
Sleeping
Sleeping
File size: 2,134 Bytes
be7c937 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 | #include <gtest/gtest.h>
#include "wayy_db/column.hpp"
using namespace wayy_db;
TEST(ColumnViewTest, BasicOperations) {
std::vector<double> data = {1.0, 2.0, 3.0, 4.0, 5.0};
ColumnView<double> view(data.data(), data.size());
EXPECT_EQ(view.size(), 5);
EXPECT_FALSE(view.empty());
EXPECT_EQ(view[0], 1.0);
EXPECT_EQ(view[4], 5.0);
EXPECT_EQ(view.front(), 1.0);
EXPECT_EQ(view.back(), 5.0);
}
TEST(ColumnViewTest, Iteration) {
std::vector<int64_t> data = {10, 20, 30};
ColumnView<int64_t> view(data.data(), data.size());
int64_t sum = 0;
for (auto val : view) {
sum += val;
}
EXPECT_EQ(sum, 60);
}
TEST(ColumnViewTest, Subview) {
std::vector<double> data = {1.0, 2.0, 3.0, 4.0, 5.0};
ColumnView<double> view(data.data(), data.size());
auto sub = view.subview(1, 3);
EXPECT_EQ(sub.size(), 3);
EXPECT_EQ(sub[0], 2.0);
EXPECT_EQ(sub[2], 4.0);
}
TEST(ColumnViewTest, OutOfRange) {
std::vector<double> data = {1.0, 2.0};
ColumnView<double> view(data.data(), data.size());
EXPECT_THROW(view.at(5), std::out_of_range);
EXPECT_THROW(view.subview(1, 5), std::out_of_range);
}
TEST(ColumnTest, ConstructWithOwnedData) {
std::vector<uint8_t> data(40); // 5 doubles
auto* ptr = reinterpret_cast<double*>(data.data());
for (int i = 0; i < 5; ++i) ptr[i] = static_cast<double>(i);
Column col("test", DType::Float64, std::move(data));
EXPECT_EQ(col.name(), "test");
EXPECT_EQ(col.dtype(), DType::Float64);
EXPECT_EQ(col.size(), 5);
EXPECT_EQ(col.byte_size(), 40);
}
TEST(ColumnTest, TypedAccess) {
std::vector<uint8_t> data(24);
auto* ptr = reinterpret_cast<int64_t*>(data.data());
ptr[0] = 100;
ptr[1] = 200;
ptr[2] = 300;
Column col("ints", DType::Int64, std::move(data));
auto view = col.as_int64();
EXPECT_EQ(view.size(), 3);
EXPECT_EQ(view[0], 100);
EXPECT_EQ(view[2], 300);
}
TEST(ColumnTest, TypeMismatch) {
std::vector<uint8_t> data(24);
Column col("ints", DType::Int64, std::move(data));
EXPECT_THROW(col.as_float64(), TypeMismatch);
}
|