test: memoryview of Apache Arrow buffers (#9)

oconnor663 · Sep 11, 2020 · 4e3a6f2 · 4e3a6f2
1 parent b69d1ba
commit 4e3a6f2
Show file tree

Hide file tree

Showing 2 changed files with 10 additions and 0 deletions.
diff --git a/.github/workflows/push.yml b/.github/workflows/push.yml
@@ -40,6 +40,8 @@ jobs:
           profile: minimal
       - name: Install pytest
         run: pip install pytest
+      - name: Install pyarrow
+        run: pip install pyarrow
       - name: Run build.py
         run: python -u tests/build.py
       - name: Run pytest

diff --git a/tests/test_blake3.py b/tests/test_blake3.py
@@ -91,6 +91,14 @@ def test_buffer_types():
     assert incremental.digest() == blake3(b"foofoofoofoo").digest()
 
 
+def test_arrow():
+    import pyarrow as pa
+    ar = pa.array(['foo', 'bar'])
+    blake = blake3()
+    blake.update(memoryview(ar.buffers()[1]))
+    blake.update(memoryview(ar.buffers()[2]))
+
+
 def test_string_fails():
     try:
         blake3("a string")