feat : Added new decorator - validate_types.

Author: manas-shinde

decorators/validate_types.py

@@ -0,0 +1,27 @@
+import functools
+import inspect
+
+
+def validate_types(func):
+    """Decorator to enforce type hints at runtime."""
+    sig = inspect.signature(func)
+    print(sig)
+
+    @functools.wraps(func)
+    def wrapper(*args, **kwargs):
+        bound = sig.bind(*args, **kwargs)
+        bound.apply_defaults()
+
+        for name, value in bound.arguments.items():
+            expected = sig.parameters[name].annotation
+            if expected is not inspect._empty and not isinstance(value, expected):
+                raise TypeError(
+                    f"Argument '{name}' must be {expected}, got {type(value)}")
+
+        result = func(*args, **kwargs)
+        if sig.return_annotation is not inspect._empty and not isinstance(result, sig.return_annotation):
+            raise TypeError(
+                f"Return value must be {sig.return_annotation}, got {type(result)}")
+        return result
+
+    return wrapper

tests/test_validate_types.py

@@ -0,0 +1,25 @@
+import pytest
+from decorators.validate_types import validate_types
+
+
+@validate_types
+def add(x: int, y: int) -> int:
+    return x + y
+
+
+def test_validate_types_success():
+    assert add(2, 3) == 5
+
+
+def test_validate_types_arg_error():
+    with pytest.raises(TypeError):
+        add("2", 3)
+
+
+def test_validate_types_return_error():
+    @validate_types
+    def broken_add(x: int, y: int) -> str:
+        return x + y  # Will return int, violates str
+
+    with pytest.raises(TypeError):
+        broken_add(1, 2)