summaryrefslogtreecommitdiffstats
path: root/src/arrow/python/pyarrow/tests/test_builder.py
diff options
context:
space:
mode:
Diffstat (limited to 'src/arrow/python/pyarrow/tests/test_builder.py')
-rw-r--r--src/arrow/python/pyarrow/tests/test_builder.py67
1 files changed, 67 insertions, 0 deletions
diff --git a/src/arrow/python/pyarrow/tests/test_builder.py b/src/arrow/python/pyarrow/tests/test_builder.py
new file mode 100644
index 000000000..50d801026
--- /dev/null
+++ b/src/arrow/python/pyarrow/tests/test_builder.py
@@ -0,0 +1,67 @@
+# Licensed to the Apache Software Foundation (ASF) under one
+# or more contributor license agreements. See the NOTICE file
+# distributed with this work for additional information
+# regarding copyright ownership. The ASF licenses this file
+# to you under the Apache License, Version 2.0 (the
+# "License"); you may not use this file except in compliance
+# with the License. You may obtain a copy of the License at
+#
+# http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing,
+# software distributed under the License is distributed on an
+# "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
+# KIND, either express or implied. See the License for the
+# specific language governing permissions and limitations
+# under the License.
+
+import weakref
+
+import numpy as np
+
+import pyarrow as pa
+from pyarrow.lib import StringBuilder
+
+
+def test_weakref():
+ sbuilder = StringBuilder()
+ wr = weakref.ref(sbuilder)
+ assert wr() is not None
+ del sbuilder
+ assert wr() is None
+
+
+def test_string_builder_append():
+ sbuilder = StringBuilder()
+ sbuilder.append(b"a byte string")
+ sbuilder.append("a string")
+ sbuilder.append(np.nan)
+ sbuilder.append(None)
+ assert len(sbuilder) == 4
+ assert sbuilder.null_count == 2
+ arr = sbuilder.finish()
+ assert len(sbuilder) == 0
+ assert isinstance(arr, pa.Array)
+ assert arr.null_count == 2
+ assert arr.type == 'str'
+ expected = ["a byte string", "a string", None, None]
+ assert arr.to_pylist() == expected
+
+
+def test_string_builder_append_values():
+ sbuilder = StringBuilder()
+ sbuilder.append_values([np.nan, None, "text", None, "other text"])
+ assert sbuilder.null_count == 3
+ arr = sbuilder.finish()
+ assert arr.null_count == 3
+ expected = [None, None, "text", None, "other text"]
+ assert arr.to_pylist() == expected
+
+
+def test_string_builder_append_after_finish():
+ sbuilder = StringBuilder()
+ sbuilder.append_values([np.nan, None, "text", None, "other text"])
+ arr = sbuilder.finish()
+ sbuilder.append("No effect")
+ expected = [None, None, "text", None, "other text"]
+ assert arr.to_pylist() == expected