Merge pull request #204 from scikit-hep/issue-203

Fixes #203.
scikit-hep · Oct 8, 2019 · 22ff4b9 · 22ff4b9
2 parents bb69702 + 4504c4c
commit 22ff4b9
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/tests/test_arrow.py b/tests/test_arrow.py
@@ -367,7 +367,10 @@ def test_arrow_nonnullable_table(self):
             x = pyarrow.array([1, 2, 3])
             y = pyarrow.array([1.1, 2.2, 3.3])
             table = pyarrow.Table.from_arrays([x], ["x"])
-            table2 = table.add_column(1, pyarrow.column(pyarrow.field("y", y.type, False), numpy.array([1.1, 2.2, 3.3])))
+            if hasattr(pyarrow, "column"):
+                table2 = table.add_column(1, pyarrow.column(pyarrow.field("y", y.type, False), numpy.array([1.1, 2.2, 3.3])))
+            else:
+                table2 = table.add_column(1, "y", y)
             assert awkward.arrow.fromarrow(table2).tolist() == [{"x": 1, "y": 1.1}, {"x": 2, "y": 2.2}, {"x": 3, "y": 3.3}]
 
     def test_arrow_trailing_zero(self):