wehub-resource-sync
29 lines
825 B
Python
29 lines
825 B
Python
from __future__ import annotations
|
|
|
|
import numpy as np
|
|
from rerun import datatypes
|
|
|
|
|
|
def test_utf8_batch_single() -> None:
|
|
single_string = "hello"
|
|
list_of_one_string = ["hello"]
|
|
array_of_one_string = np.array(["hello"])
|
|
|
|
assert (
|
|
datatypes.Utf8Batch(single_string).as_arrow_array() == datatypes.Utf8Batch(list_of_one_string).as_arrow_array()
|
|
)
|
|
|
|
assert (
|
|
datatypes.Utf8Batch(single_string).as_arrow_array() == datatypes.Utf8Batch(array_of_one_string).as_arrow_array()
|
|
)
|
|
|
|
|
|
def test_utf8_batch_many() -> None:
|
|
# different string length to be sure
|
|
list_of_strings = ["hell", "worlds"]
|
|
array_of_strings = np.array(["hell", "worlds"])
|
|
|
|
assert (
|
|
datatypes.Utf8Batch(list_of_strings).as_arrow_array() == datatypes.Utf8Batch(array_of_strings).as_arrow_array()
|
|
)
|