Working on improving specificity of type hinting

Author: allRisc

.github/workflows/ci.yml

@@ -39,6 +39,8 @@ jobs:
         run: uv run pytest -v
 
       - name: Run StubTest
+        env:
+          MYPYPATH: ./stubs
         run: uv run stubtest pyslang --allowlist allowlist.txt
 
       - name: Test Package Build

src/pyslang-stubs/__init__.pyi

@@ -173,7 +173,7 @@ class ASTContext(metaclass=_metaclass):
     def evalInteger(self, expr: Any, extraFlags: EvalFlags = ...) -> int | None: ...
     def evalPackedDimension(self, syntax: Any) -> EvaluatedDimension: ...
     def evalUnpackedDimension(self, syntax: Any) -> EvaluatedDimension: ...
-    def getRandMode(self, symbol: Any) -> Any: ...
+    def getRandMode(self, symbol: Any) -> RandMode: ...
     def requireBooleanConvertible(self, expr: Any) -> bool: ...
     def requireGtZero(self, value: int | None, range: Any) -> bool: ...
     @typing.overload
@@ -183,33 +183,33 @@ class ASTContext(metaclass=_metaclass):
     def requireNoUnknowns(self, value: Any, range: Any) -> bool: ...
     def requirePositive(self, value: Any, range: Any) -> bool: ...
     @typing.overload
-    def requireSimpleExpr(self, expr: Any) -> Any: ...
+    def requireSimpleExpr(self, expr: Any) -> ExpressionSyntax: ...
     @typing.overload
-    def requireSimpleExpr(self, expr: Any, code: Any) -> Any: ...
+    def requireSimpleExpr(self, expr: Any, code: Any) -> ExpressionSyntax: ...
     @typing.overload
     def requireValidBitWidth(self, width: int, range: Any) -> bool: ...
     @typing.overload
     def requireValidBitWidth(self, value: Any, range: Any) -> int | None: ...
     def resetFlags(self, addedFlags: ASTFlags) -> ASTContext: ...
-    def tryEval(self, expr: Any) -> Any: ...
+    def tryEval(self, expr: Any) -> ConstantValue: ...
     @property
     def flags(self) -> ASTFlags: ...
     @property
-    def getCompilation(self) -> Any: ...
+    def getCompilation(self) -> Compilation: ...
     @property
-    def getInstance(self) -> Any: ...
+    def getInstance(self) -> InstanceSymbolBase: ...
     @property
-    def getLocation(self) -> Any: ...
+    def getLocation(self) -> LookupLocation: ...
     @property
-    def getProceduralBlock(self) -> Any: ...
+    def getProceduralBlock(self) -> ProceduralBlockSymbol: ...
     @property
     def inAlwaysCombLatch(self) -> bool: ...
     @property
     def inUnevaluatedBranch(self) -> bool: ...
     @property
     def lookupIndex(self) -> Any: ...
     @property
-    def scope(self) -> Any: ...
+    def scope(self) -> Scope: ...
 
 class AbortAssertionExpr(AssertionExpr):
     class Action(metaclass=_metaclass):
@@ -231,9 +231,9 @@ class AbortAssertionExpr(AssertionExpr):
     Reject: Final = Action.Reject
 
     @property
-    def action(self) -> Any: ...
+    def action(self) -> Action: ...
     @property
-    def condition(self) -> Any: ...
+    def condition(self) -> Expression: ...
     @property
     def expr(self) -> AssertionExpr: ...
     @property
@@ -1596,7 +1596,7 @@ class ConstantValue(metaclass=_metaclass):
     def isTrue(self) -> bool: ...
     def size(self) -> int: ...
     @property
-    def value(self) -> typing.Any: ...
+    def value(self) -> Any: ...
 
 class Constraint(metaclass=_metaclass):
     def __repr__(self) -> str: ...