Added more tests

Author: ankane

.github/workflows/build.yml

@@ -22,18 +22,18 @@ jobs:
           make
           sudo make install
 
-      - run: g++ -std=c++17 -Wall -Wextra -Werror -o test/pqxx test/pqxx_test.cpp -lpqxx -lpq
-      - run: test/pqxx
+      - run: g++ -std=c++17 -Wall -Wextra -Werror -o test/main test/main.cpp test/pqxx_test.cpp -lpqxx -lpq
+      - run: test/main
 
-      - run: g++ -std=c++20 -Wall -Wextra -Werror -o test/pqxx test/pqxx_test.cpp -lpqxx -lpq
-      - run: test/pqxx
+      - run: g++ -std=c++20 -Wall -Wextra -Werror -o test/main test/main.cpp test/pqxx_test.cpp -lpqxx -lpq
+      - run: test/main
 
-      - run: g++ -std=c++23 -Wall -Wextra -Werror -o test/pqxx test/pqxx_test.cpp -lpqxx -lpq
-      - run: test/pqxx
+      - run: g++ -std=c++23 -Wall -Wextra -Werror -o test/main test/main.cpp test/pqxx_test.cpp -lpqxx -lpq
+      - run: test/main
 
       - run: |
           sudo apt-get install valgrind
-          valgrind --leak-check=yes test/pqxx
+          valgrind --leak-check=yes test/main
 
       # test install
       - run: cmake -S . -B build

.gitignore

@@ -1,2 +1,2 @@
 build
-/test/pqxx
+/test/main

README.md

@@ -107,8 +107,8 @@ To get started with development:
 git clone https://github.com/pgvector/pgvector-cpp.git
 cd pgvector-cpp
 createdb pgvector_cpp_test
-g++ -std=c++17 -Wall -Wextra -Wno-unknown-attributes -Werror -o test/pqxx test/pqxx_test.cpp -lpqxx -lpq
-test/pqxx
+g++ -std=c++17 -Wall -Wextra -Wno-unknown-attributes -Werror -o test/main test/main.cpp test/pqxx_test.cpp -lpqxx -lpq
+test/main
 ```
 
 To run an example:

test/main.cpp

@@ -0,0 +1,34 @@
+#include <cassert>
+
+#include "../include/pgvector/halfvec.hpp"
+#include "../include/pgvector/sparsevec.hpp"
+#include "../include/pgvector/vector.hpp"
+
+using pgvector::HalfVector;
+using pgvector::SparseVector;
+using pgvector::Vector;
+
+void test_pqxx();
+
+void test_vector() {
+    auto vec = Vector({1, 2, 3});
+    assert(vec.dimensions() == 3);
+}
+
+void test_halfvec() {
+    auto vec = HalfVector({1, 2, 3});
+    assert(vec.dimensions() == 3);
+}
+
+void test_sparsevec() {
+    auto vec = SparseVector({1, 2, 3});
+    assert(vec.dimensions() == 3);
+}
+
+int main() {
+    test_pqxx();
+    test_vector();
+    test_halfvec();
+    test_sparsevec();
+    return 0;
+}

test/pqxx_test.cpp

@@ -1,8 +1,9 @@
-#include "../include/pgvector/pqxx.hpp"
 #include <cassert>
 #include <optional>
 #include <pqxx/pqxx>
 
+#include "../include/pgvector/pqxx.hpp"
+
 void setup(pqxx::connection &conn) {
     pqxx::nontransaction tx(conn);
     tx.exec("CREATE EXTENSION IF NOT EXISTS vector");
@@ -131,7 +132,7 @@ void test_precision(pqxx::connection &conn) {
     assert(res[0][0].as<pgvector::Vector>() == embedding);
 }
 
-int main() {
+void test_pqxx() {
     pqxx::connection conn("dbname=pgvector_cpp_test");
     setup(conn);
 
@@ -143,6 +144,4 @@ int main() {
     test_stream(conn);
     test_stream_to(conn);
     test_precision(conn);
-
-    return 0;
 }