Function test_struct_meta_subclass_functions

tests/unit/test_struct_meta.py:237–295

Test if structs created by StructMeta subclasses support various function operations.

()

def test_struct_meta_subclass_functions():
    """Test if structs created by StructMeta subclasses support various function operations."""

    # Define a custom metaclass
    class CustomMeta(StructMeta):
        """Custom metaclass that inherits from StructMeta"""

    # Use the custom metaclass to create a struct class
    class CustomStruct(metaclass=CustomMeta):
        x: int
        y: str
        z: float = 3.14

    # Create an instance
    obj = CustomStruct(x=1, y="test")
    assert obj.x == 1
    assert obj.y == "test"
    assert obj.z == 3.14

    # Test asdict function
    d = asdict(obj)
    assert isinstance(d, dict)
    assert d["x"] == 1
    assert d["y"] == "test"
    assert d["z"] == 3.14

    # Test astuple function
    t = astuple(obj)
    assert isinstance(t, tuple)
    assert t == (1, "test", 3.14)

    # Test replace function
    obj2 = replace(obj, y="replaced")
    assert obj2.x == 1
    assert obj2.y == "replaced"
    assert obj2.z == 3.14

    # Test force_setattr function
    force_setattr(obj, "x", 100)
    assert obj.x == 100

    # Test nested structs
    class NestedStruct(metaclass=CustomMeta):
        inner: CustomStruct
        name: str

    nested = NestedStruct(inner=obj, name="nested")
    assert nested.inner.x == 100
    assert nested.inner.y == "test"
    assert nested.name == "nested"

    # Test asdict with nested structs
    nested_dict = asdict(nested)
    assert isinstance(nested_dict, dict)
    # Note: asdict doesn't recursively convert nested struct objects, so inner remains a CustomStruct object
    assert isinstance(nested_dict["inner"], CustomStruct)
    assert nested_dict["inner"].x == 100
    assert nested_dict["inner"].y == "test"