Make equality operator work with nulls (#24)

pgvector · Oct 16, 2023 · ae69b85 · ae69b85
1 parent 59ddad1
commit ae69b85
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 2 deletions.
diff --git a/src/Pgvector/Vector.cs b/src/Pgvector/Vector.cs
@@ -26,8 +26,10 @@ public bool Equals(Vector? other)
     public override bool Equals(object? obj)
         => obj is Vector vector && Equals(vector);
 
-    public static bool operator ==(Vector x, Vector y) => x.Equals(y);
-    public static bool operator !=(Vector x, Vector y) => !(x == y);
+    public static bool operator ==(Vector? x, Vector? y)
+        => x is null && y is null || x is not null && x.Equals(y);
+
+    public static bool operator !=(Vector? x, Vector? y) => !(x == y);
 
     public override int GetHashCode()
     {

diff --git a/tests/Pgvector.Tests/VectorTests.cs b/tests/Pgvector.Tests/VectorTests.cs
@@ -19,5 +19,9 @@ public void Equal()
 
         Assert.False(a != b);
         Assert.True(a != c);
+
+        Assert.False(a == null);
+        Assert.False(null == a);
+        Assert.True((Vector?)null == null);
     }
 }