Added test for binary copy

pgvector · Mar 6, 2024 · 377a419 · 377a419
1 parent ada5859
commit 377a419
Showing 1 changed file with 14 additions and 0 deletions.
diff --git a/tests/Pgvector.CSharp.Tests/NpgsqlTests.cs b/tests/Pgvector.CSharp.Tests/NpgsqlTests.cs
@@ -69,6 +69,20 @@ public async Task Main()
             await cmd.ExecuteNonQueryAsync();
         }
 
+        await using (var writer = conn.BeginBinaryImport("COPY items (embedding) FROM STDIN WITH (FORMAT BINARY)"))
+        {
+            writer.StartRow();
+            writer.Write(new Vector(new float[] { 1, 1, 1 }));
+
+            writer.StartRow();
+            writer.Write(new Vector(new float[] { 2, 2, 2 }));
+
+            writer.StartRow();
+            writer.Write(new Vector(new float[] { 1, 1, 2 }));
+
+            writer.Complete();
+        }
+
         await using (var cmd = new NpgsqlCommand("SELECT $1", conn))
         {
             var embedding = new Vector(new float[16000]);