test: test split da

jina-ai · Aug 4, 2022 · 6220a2a · 6220a2a
1 parent b03d330
commit 6220a2a
Showing 1 changed file with 19 additions and 0 deletions.
diff --git a/tests/test_helper.py b/tests/test_helper.py
@@ -40,6 +40,25 @@ def test_numpy_softmax(shape, axis):
             ),
             (3, 1),
         ),
+        (
+            DocumentArray(
+                [
+                    Document(text='hello, world'),
+                    Document(tensor=np.array([0, 1, 2])),
+                    Document(
+                        uri='https://docarray.jina.ai/_static/favicon.png'
+                    ).load_uri_to_blob(),
+                    Document(
+                        tensor=np.array([0, 1, 2]),
+                        uri='https://docarray.jina.ai/_static/favicon.png',
+                    ),
+                    Document(
+                        uri='https://docarray.jina.ai/_static/favicon.png',
+                    ),
+                ]
+            ),
+            (1, 4),
+        ),
         (
             DocumentArray(
                 [