docarray
diff --git a/Collapse file
‎docarray/array/doc_vec/column_storage.py‎
Copy file name to clipboardExpand all lines: docarray/array/doc_vec/column_storage.py
+15Lines changed: 15 additions & 0 deletions b/Collapse file
‎docarray/array/doc_vec/column_storage.py‎
Copy file name to clipboardExpand all lines: docarray/array/doc_vec/column_storage.py
+15Lines changed: 15 additions & 0 deletions
diff --git a/Collapse file
‎docarray/array/doc_vec/doc_vec.py‎
Copy file name to clipboardExpand all lines: docarray/array/doc_vec/doc_vec.py
+11Lines changed: 11 additions & 0 deletions b/Collapse file
‎docarray/array/doc_vec/doc_vec.py‎
Copy file name to clipboardExpand all lines: docarray/array/doc_vec/doc_vec.py
+11Lines changed: 11 additions & 0 deletions
diff --git a/Collapse file
‎tests/units/array/stack/test_array_stacked.py‎
Copy file name to clipboardExpand all lines: tests/units/array/stack/test_array_stacked.py
+32Lines changed: 32 additions & 0 deletions b/Collapse file
‎tests/units/array/stack/test_array_stacked.py‎
Copy file name to clipboardExpand all lines: tests/units/array/stack/test_array_stacked.py
+32Lines changed: 32 additions & 0 deletions
@@ -91,6 +91,21 @@ def __getitem__(self: T, item: IndexIterType) -> T:
             self.tensor_type,
         )
 
+    def __eq__(self, other: Any) -> bool:
+        if not isinstance(other, ColumnStorage):
+            return False
+        if self.tensor_type != other.tensor_type:
+            return False
+        for col_map_self, col_map_other in zip(self.columns.maps, other.columns.maps):
+            if col_map_self.keys() != col_map_other.keys():
+                return False
+            for key_self in col_map_self.keys():
+                if key_self == 'id':
+                    continue
+                if col_map_self[key_self] != col_map_other[key_self]:
+                    return False
+        return True
+
 
 class ColumnStorageView(dict, MutableMapping[str, Any]):
     index: int
 
@@ -584,6 +584,17 @@ def __iter__(self):
     def __len__(self):
         return len(self._storage)
 
+    def __eq__(self, other: Any) -> bool:
+        if not isinstance(other, DocVec):
+            return False
+        if self.doc_type != other.doc_type:
+            return False
+        if self.tensor_type != other.tensor_type:
+            return False
+        if self._storage != other._storage:
+            return False
+        return True
+
     ####################
     # IO related       #
     ####################
 
@@ -585,3 +585,35 @@ def test_doc_view_dict(batch):
     d = doc_view_two.dict()
     assert d['tensor'].shape == (3, 224, 224)
     assert d['id'] == doc_view_two.id
+
+
+def test_doc_vec_equality():
+    class Text(BaseDoc):
+        text: str
+
+    da = DocVec[Text]([Text(text='hello') for _ in range(10)])
+    da2 = DocList[Text]([Text(text='hello') for _ in range(10)])
+
+    assert da != da2
+    assert da == da2.to_doc_vec()
+
+
+def test_doc_vec_nested(batch_nested_doc):
+    batch, Doc, Inner = batch_nested_doc
+    batch2 = DocVec[Doc]([Doc(inner=Inner(hello='hello')) for _ in range(10)])
+
+    assert batch == batch2
+
+
+def test_doc_vec_tensor_type():
+    class ImageDoc(BaseDoc):
+        tensor: AnyTensor
+
+    da = DocVec[ImageDoc]([ImageDoc(tensor=np.zeros((3, 224, 224))) for _ in range(10)])
+
+    da2 = DocVec[ImageDoc](
+        [ImageDoc(tensor=torch.zeros(3, 224, 224)) for _ in range(10)],
+        tensor_type=TorchTensor,
+    )
+
+    assert da != da2