wehub-resource-sync
91 lines
2.1 KiB
Python
91 lines
2.1 KiB
Python
import os
|
|
import sys
|
|
|
|
import pytest
|
|
|
|
from ray import (
|
|
ActorClassID,
|
|
ActorID,
|
|
ClusterID,
|
|
FunctionID,
|
|
JobID,
|
|
NodeID,
|
|
PlacementGroupID,
|
|
TaskID,
|
|
WorkerID,
|
|
)
|
|
|
|
|
|
@pytest.mark.parametrize(
|
|
"id_cls, size",
|
|
[
|
|
(JobID, 4),
|
|
(ActorID, 16),
|
|
(TaskID, 24),
|
|
(NodeID, 28),
|
|
(WorkerID, 28),
|
|
(FunctionID, 28),
|
|
(ActorClassID, 28),
|
|
(ClusterID, 28),
|
|
(PlacementGroupID, 18),
|
|
],
|
|
)
|
|
def test_id_methods(id_cls, size):
|
|
"""
|
|
Tests for ID class methods:
|
|
|
|
Methods:
|
|
- `ctor`
|
|
- `from_binary(bytes: bytes) -> id_cls`
|
|
- `binary() -> bytes`
|
|
- `hex() -> str`
|
|
- `from_hex(hex_str: str) -> id_cls`
|
|
- `size(cls) -> int`
|
|
- `nil() -> id_cls`
|
|
- `is_nil() -> bool`
|
|
- `__hash__() -> int`
|
|
|
|
Invariants:
|
|
- `ctor` is same as `from_binary`
|
|
- `id_cls.from_binary(id.binary()) == id`
|
|
- `id_cls.from_hex(id.hex()) == id`
|
|
- `id_cls.size() == expected_size`
|
|
- `id_cls.nil().binary() == b'\xff' * expected_size`
|
|
- `id_cls.nil().hex() == 'ff' * expected_size`
|
|
- `id_cls.nil().is_nil() is True`
|
|
|
|
Note: we don't test ObjectID (that is, ObjectRef) here. It lacks from_binary and
|
|
from_hex methods.
|
|
"""
|
|
# Generate random bytes of the given size
|
|
random_bytes = os.urandom(size)
|
|
|
|
id_instance = id_cls(random_bytes)
|
|
|
|
# binary and from_binary
|
|
id_from_binary = id_cls.from_binary(random_bytes)
|
|
assert id_instance == id_from_binary
|
|
assert id_from_binary.binary() == random_bytes
|
|
|
|
# hex and from_hex methods
|
|
hex_str = id_instance.hex()
|
|
assert isinstance(hex_str, str)
|
|
assert id_cls.from_hex(hex_str) == id_instance
|
|
|
|
# size() is classmethod, can be called on both class and instance
|
|
assert id_cls.size() == size
|
|
assert id_instance.size() == size
|
|
|
|
# Test the nil method
|
|
nil_instance = id_cls.nil()
|
|
assert nil_instance.binary() == b"\xff" * size
|
|
assert nil_instance.hex() == "ff" * size
|
|
assert nil_instance.is_nil()
|
|
|
|
# hash is implemented
|
|
id_instance.__hash__() is not None
|
|
|
|
|
|
if __name__ == "__main__":
|
|
sys.exit(pytest.main(["-sv", __file__]))
|