pyarrow-assert 0.3.6__tar.gz → 0.3.7__tar.gz

{pyarrow_assert-0.3.6 → pyarrow_assert-0.3.7}/PKG-INFO

@@ -1,24 +1,21 @@
 Metadata-Version: 2.1
 Name: pyarrow-assert
-Version: 0.3.6
+Version: 0.3.7
 Summary: Dora Node for plotting text and bbox on image with OpenCV
 Home-page: https://github.com/dora-rs/dora.git
 License: MIT
 Author: Haixuan Xavier Tao
 Author-email: tao.xavier@outlook.com
+Requires-Python: >=3.7,<4.0
 Classifier: License :: OSI Approved :: MIT License
-Classifier: Programming Language :: Python :: 2
-Classifier: Programming Language :: Python :: 2.7
 Classifier: Programming Language :: Python :: 3
-Classifier: Programming Language :: Python :: 3.4
-Classifier: Programming Language :: Python :: 3.5
-Classifier: Programming Language :: Python :: 3.6
 Classifier: Programming Language :: Python :: 3.7
 Classifier: Programming Language :: Python :: 3.8
 Classifier: Programming Language :: Python :: 3.9
 Classifier: Programming Language :: Python :: 3.10
 Classifier: Programming Language :: Python :: 3.11
 Classifier: Programming Language :: Python :: 3.12
+Classifier: Programming Language :: Python :: 3.13
 Requires-Dist: dora-rs (>=0.3.6,<0.4.0)
 Requires-Dist: numpy (<2.0.0)
 Requires-Dist: pyarrow (>=5.0.0)

{pyarrow_assert-0.3.6 → pyarrow_assert-0.3.7}/pyarrow_assert/main.py

@@ -2,7 +2,7 @@ import argparse
 import os
 import ast
 
-
+import pyarrow as pa
 from dora import Node
 
 RUNNER_CI = True if os.getenv("CI") == "true" else False
@@ -36,12 +36,23 @@ def main():
         args.name
     )  # provide the name to connect to the dataflow if dynamic node
 
-    assert_data = ast.literal_eval(data)
+    data = ast.literal_eval(data)
+
+    if isinstance(data, list):
+        data = pa.array(data)  # initialize pyarrow array
+    elif isinstance(data, str):
+        data = pa.array([data])
+    elif isinstance(data, int):
+        data = pa.array([data])
+    elif isinstance(data, float):
+        data = pa.array([data])
+    else:
+        data = pa.array(data)  # initialize pyarrow array
 
     for event in node:
         if event["type"] == "INPUT":
             value = event["value"]
-            assert value == assert_data, f"Expected {assert_data}, got {value}"
+            assert value == data, f"Expected {data}, got {value}"
 
 
 if __name__ == "__main__":

{pyarrow_assert-0.3.6 → pyarrow_assert-0.3.7}/pyproject.toml

@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "pyarrow-assert"
-version = "0.3.6"
+version = "0.3.7"
 authors = [
     "Haixuan Xavier Tao <tao.xavier@outlook.com>",
     "Enzo Le Van <dev@enzo-le-van.fr>",
@@ -16,6 +16,7 @@ packages = [{ include = "pyarrow_assert" }]
 dora-rs = "^0.3.6"
 numpy = "< 2.0.0"
 pyarrow = ">= 5.0.0"
+python = "^3.7"
 
 [tool.poetry.scripts]
 pyarrow-assert = "pyarrow_assert.main:main"
@@ -23,6 +24,3 @@ pyarrow-assert = "pyarrow_assert.main:main"
 [build-system]
 requires = ["poetry-core>=1.8.0"]
 build-backend = "poetry.core.masonry.api"
-
-[project]
-readme = "README.md"