Added more tests [skip ci]

pgvector · Jun 9, 2024 · 33eb182 · 33eb182
1 parent b3851e5
commit 33eb182
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/tests/test_sparse_vector.py b/tests/test_sparse_vector.py
@@ -16,6 +16,8 @@ def test_from_coordinates(self):
     def test_from_sparse(self):
         arr = coo_array(np.array([1, 0, 2, 0, 3, 0]))
         assert SparseVector.from_sparse(arr).to_list() == [1, 0, 2, 0, 3, 0]
+        assert SparseVector.from_sparse(arr.tocsc()).to_list() == [1, 0, 2, 0, 3, 0]
+        assert SparseVector.from_sparse(arr.tocsr()).to_list() == [1, 0, 2, 0, 3, 0]
         assert SparseVector.from_sparse(arr.todok()).to_list() == [1, 0, 2, 0, 3, 0]
 
     def test_repr(self):