def test_time32_python(self):
"""
Python -> Rust -> Python
"""
a = pyarrow.array([None, 1, 2], pyarrow.time32('s'))
b = arrow_pyarrow_integration_testing.concatenate(a)
expected = pyarrow.array([None, 1, 2] + [None, 1, 2], pyarrow.time32('s'))
self.assertEqual(b, expected)
def test_time32_python():
"""
Python -> Rust -> Python
"""
a = pa.array([None, 1, 2], pa.time32("s"))
b = rust.concatenate(a)
expected = pa.array([None, 1, 2] + [None, 1, 2], pa.time32("s"))
assert b == expected
del a
del b
del expected
def test_time32_python(self):
"""
Python -> Rust -> Python
"""
old_allocated = pyarrow.total_allocated_bytes()
a = pyarrow.array([None, 1, 2], pyarrow.time32('s'))
b = arrow_pyarrow_integration_testing.concatenate(a)
expected = pyarrow.array([None, 1, 2] + [None, 1, 2], pyarrow.time32('s'))
self.assertEqual(b, expected)
del a
del b
del expected
# No leak of C++ memory
self.assertEqual(old_allocated, pyarrow.total_allocated_bytes())
def test_timestamp_python():
"""
Python -> Rust -> Python
"""
data = [
None,
datetime.datetime(2021, 1, 1, 1, 1, 1, 1),
datetime.datetime(2020, 3, 9, 1, 1, 1, 1),
]
a = pa.array(data, pa.timestamp("us"))
b = rust.concatenate(a)
expected = pa.array(data + data, pa.timestamp("us"))
assert b == expected
del a
del b
del expected
def test_timestamp_tz_python():
"""
Python -> Rust -> Python
"""
tzinfo = pytz.timezone("America/New_York")
pyarrow_type = pa.timestamp("us", tz="America/New_York")
data = [
None,
datetime.datetime(2021, 1, 1, 1, 1, 1, 1, tzinfo=tzinfo),
datetime.datetime(2020, 3, 9, 1, 1, 1, 1, tzinfo=tzinfo),
]
a = pa.array(data, type=pyarrow_type)
b = rust.concatenate(a)
expected = pa.array(data * 2, type=pyarrow_type)
assert b == expected
del a
del b
del expected
