Skip to content

Commit 2fdb473

Browse files
committed
Improved SparseVector tests
1 parent a8a63ba commit 2fdb473

File tree

1 file changed

+8
-0
lines changed

1 file changed

+8
-0
lines changed

tests/Pgvector.CSharp.Tests/SparseVectorTests.cs

Lines changed: 8 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -10,13 +10,19 @@ public void StringConstructor()
1010
var v = new SparseVector("{1:1,3:2,5:3}/6");
1111
Assert.Equal("{1:1,3:2,5:3}/6", v.ToString());
1212
Assert.Equal(new float[] { 1, 0, 2, 0, 3, 0 }, v.ToArray());
13+
Assert.Equal(6, v.Dimensions);
14+
Assert.Equal(new int[] { 0, 2, 4 }, v.Indices.ToArray());
15+
Assert.Equal(new float[] { 1, 2, 3 }, v.Values.ToArray());
1316
}
1417

1518
[Fact]
1619
public void ArrayConstructor()
1720
{
1821
var v = new SparseVector(new float[] { 1, 0, 2, 0, 3, 0 });
1922
Assert.Equal(new float[] { 1, 0, 2, 0, 3, 0 }, v.ToArray());
23+
Assert.Equal(6, v.Dimensions);
24+
Assert.Equal(new int[] { 0, 2, 4 }, v.Indices.ToArray());
25+
Assert.Equal(new float[] { 1, 2, 3 }, v.Values.ToArray());
2026
}
2127

2228
[Fact]
@@ -29,7 +35,9 @@ public void DictionaryConstructor()
2935
dictionary.Add(3, 0);
3036
var v = new SparseVector(dictionary, 6);
3137
Assert.Equal(new float[] { 1, 0, 2, 0, 3, 0 }, v.ToArray());
38+
Assert.Equal(6, v.Dimensions);
3239
Assert.Equal(new int[] { 0, 2, 4 }, v.Indices.ToArray());
40+
Assert.Equal(new float[] { 1, 2, 3 }, v.Values.ToArray());
3341
}
3442

3543
[Fact]

0 commit comments

Comments
 (0)