test: align stub checks with non-deprecated constants

Author: banteg

tests/test_pyray_stub_parity.py

@@ -31,43 +31,32 @@ def _top_level_names(module: ast.Module) -> set[str]:
     return names
 
 
-def _enum_member_names(module: ast.Module) -> set[str]:
-    names: set[str] = set()
-    for node in module.body:
-        if not isinstance(node, ast.ClassDef):
-            continue
-        if not any(isinstance(base, ast.Name) and base.id == "int" for base in node.bases):
-            continue
-        for class_node in node.body:
-            if not isinstance(class_node, ast.Assign):
-                continue
-            for target in class_node.targets:
-                if isinstance(target, ast.Name) and target.id.isupper():
-                    names.add(target.id)
-    return names
-
-
-def test_enum_members_are_available_as_module_constants() -> None:
-    module = _parse_stub()
-    missing = sorted(_enum_member_names(module) - _top_level_names(module))
-    assert not missing, (
-        "Missing module-level constants for enum members; "
-        f"first missing names: {missing[:15]}"
-    )
-
-
 def test_stub_includes_expected_smoke_constants() -> None:
     module_names = _top_level_names(_parse_stub())
     required_names = {
-        "BLEND_ALPHA",
-        "MOUSE_BUTTON_LEFT",
+        "RL_FUNC_ADD",
+        "RL_ONE",
+        "RL_ONE_MINUS_SRC_ALPHA",
         "RL_QUADS",
         "RL_SRC_ALPHA",
-        "TEXTURE_FILTER_POINT",
+        "RL_ZERO",
     }
     assert required_names.issubset(module_names)
 
 
+def test_stub_excludes_deprecated_shorthand_constants() -> None:
+    module_names = _top_level_names(_parse_stub())
+    omitted_names = {
+        "ARROWS_SIZE",
+        "BLEND_ALPHA",
+        "MOUSE_BUTTON_LEFT",
+        "SHADER_LOC_MATRIX_MVP",
+        "SHADER_UNIFORM_FLOAT",
+        "TEXTURE_FILTER_POINT",
+    }
+    assert omitted_names.isdisjoint(module_names)
+
+
 def test_texture2d_is_a_texture_alias() -> None:
     module = _parse_stub()
     class_names = {

tests/typing/pyray_stub_smoke.py

@@ -1,10 +1,11 @@
 import pyray as rl
 
 
-blend: int = rl.BLEND_ALPHA
+blend: int = rl.BlendMode.BLEND_ALPHA
 mode: int = rl.RL_QUADS
-mouse: int = rl.MOUSE_BUTTON_LEFT
-flt: int = rl.TEXTURE_FILTER_POINT
+src_alpha: int = rl.RL_SRC_ALPHA
+mouse: int = rl.MouseButton.MOUSE_BUTTON_LEFT
+flt: int = rl.TextureFilter.TEXTURE_FILTER_POINT
 
 
 def accepts_texture(texture: rl.Texture | None) -> None: