Address review comments

dissect/cstruct/compiler.py

@@ -407,7 +407,7 @@ def _generate_struct_info(cs: cstruct, fields: list[Field], align: bool = False)
             current_offset += size
 
 
-def _optimize_struct_fmt(info: Iterator[tuple[Field, int, str]]):
+def _optimize_struct_fmt(info: Iterator[tuple[Field, int, str]]) -> str:
     chars = []
 
     current_count = 0

dissect/cstruct/types/char.py

@@ -15,7 +15,7 @@ def _read(cls, stream: BinaryIO, context: dict[str, Any] = None) -> Char:
     @classmethod
     def _read_array(cls, stream: BinaryIO, count: int, context: dict[str, Any] = None) -> Char:
         if count == 0:
-            return b""
+            return type.__call__(cls, b"")
 
         data = stream.read(-1 if count == EOF else count)
         if count != EOF and len(data) != count:

dissect/cstruct/types/enum.py

@@ -72,6 +72,7 @@ def _write_0(cls, stream: BinaryIO, array: list[BaseType]) -> int:
 
 def _fix_alias_members(cls: type[Enum]):
     # Emulate aenum NoAlias behaviour
+    # https://github.com/ethanfurman/aenum/blob/master/aenum/doc/aenum.rst
     if len(cls._member_names_) == len(cls._member_map_):
         return
 

dissect/cstruct/types/structure.py

@@ -457,9 +457,7 @@ class Union(Structure, metaclass=UnionMetaType):
     """Base class for cstruct union type classes."""
 
     def __eq__(self, other: Any) -> bool:
-        if self.__class__ is other.__class__:
-            return bytes(self) == bytes(other)
-        return False
+        return self.__class__ is other.__class__ and bytes(self) == bytes(other)
 
 
 def _codegen(func: FunctionType) -> FunctionType:

dissect/cstruct/types/wchar.py

@@ -24,7 +24,7 @@ def _read(cls, stream: BinaryIO, context: dict[str, Any] = None) -> Wchar:
     @classmethod
     def _read_array(cls, stream: BinaryIO, count: int, context: dict[str, Any] = None) -> Wchar:
         if count == 0:
-            return ""
+            return type.__call__(cls, "")
 
         if count != EOF:
             count *= 2
@@ -40,8 +40,8 @@ def _read_0(cls, stream: BinaryIO, context: dict[str, Any] = None) -> Wchar:
         buf = []
         while True:
             point = stream.read(2)
-            if len(point) != 2:
-                raise EOFError("Read 0 bytes, but expected 2")
+            if (bytes_read := len(point)) != 2:
+                raise EOFError(f"Read {bytes_read} bytes, but expected 2")
 
             if point == b"\x00\x00":
                 break