chore: import upstream snapshot with attribution

python/ray/tests/test_ids.py

@@ -0,0 +1,90 @@
+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__]))