add sanity checks

vyperlang · Jan 14, 2024 · 9e60d60 · 9e60d60
1 parent 9fe84f8
commit 9e60d60
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 0 deletions.
diff --git a/vyper/codegen/function_definitions/external_function.py b/vyper/codegen/function_definitions/external_function.py
@@ -171,6 +171,8 @@ def generate_ir_for_external_function(code, compilation_target):
     metadata required for `module.py` to construct the selector table.
     """
     func_t = code._metadata["func_type"]
+    assert func_t.is_external or func_t.is_constructor  # sanity check
+
     context = initialize_context(func_t, compilation_target, func_t.is_constructor)
     nonreentrant_pre, nonreentrant_post = get_nonreentrant_lock(func_t)
 

diff --git a/vyper/codegen/function_definitions/internal_function.py b/vyper/codegen/function_definitions/internal_function.py
@@ -41,6 +41,9 @@ def generate_ir_for_internal_function(
 
     func_t = code._metadata["func_type"]
 
+    # sanity check
+    assert func_t.is_internal or func_t.is_constructor
+
     context = initialize_context(func_t, module_ctx, is_ctor_context)
 
     for arg in func_t.arguments: