wehub-resource-sync
49 lines
1.2 KiB
Python
49 lines
1.2 KiB
Python
from __future__ import annotations
|
|
|
|
import itertools
|
|
from typing import cast
|
|
|
|
import numpy as np
|
|
import rerun as rr
|
|
import rerun.blueprint as rrb
|
|
|
|
from .common_arrays import none_empty_or_value
|
|
|
|
|
|
def test_scalar_axis() -> None:
|
|
rr.set_strict_mode(True)
|
|
|
|
# All from 42.1337 to 1337.42, but expressed differently
|
|
ranges = [
|
|
(42.1337, 1337.42),
|
|
[42.1337, 1337.42],
|
|
np.array([42.1337, 1337.42]),
|
|
rr.components.Range1D([42.1337, 1337.42]),
|
|
None,
|
|
]
|
|
zoom_locks = [
|
|
True,
|
|
False,
|
|
]
|
|
|
|
all_arrays = itertools.zip_longest(
|
|
ranges,
|
|
zoom_locks,
|
|
)
|
|
|
|
for range, zoom_lock in all_arrays:
|
|
range = cast("rr.datatypes.Range1DLike | None", range)
|
|
zoom_lock = cast("rr.datatypes.Bool | None", zoom_lock)
|
|
|
|
print(
|
|
f"rr.ScalarAxis(\n range={range!r}\n zoom_lock={zoom_lock!r}\n)",
|
|
)
|
|
arch = rrb.ScalarAxis(
|
|
range=range,
|
|
zoom_lock=zoom_lock,
|
|
)
|
|
print(f"{arch}\n")
|
|
|
|
assert arch.range == rr.components.Range1DBatch._converter(none_empty_or_value(range, [42.1337, 1337.42]))
|
|
assert arch.zoom_lock == rrb.components.LockRangeDuringZoomBatch._converter(zoom_lock)
|