41 lines
1.6 KiB
Python
41 lines
1.6 KiB
Python
|
"""Test that C functions used in primitives are declared in a header such as CPy.h."""
|
||
|
|
|
||
|
|
import glob
|
||
|
|
import os
|
||
|
|
import re
|
||
|
|
import unittest
|
||
|
|
|
||
|
|
from mypyc.primitives import registry
|
||
|
|
from mypyc.primitives.registry import CFunctionDescription
|
||
|
|
|
||
|
|
|
||
|
|
class TestHeaderInclusion(unittest.TestCase):
|
||
|
|
def test_primitives_included_in_header(self) -> None:
|
||
|
|
base_dir = os.path.join(os.path.dirname(__file__), '..', 'lib-rt')
|
||
|
|
with open(os.path.join(base_dir, 'CPy.h')) as f:
|
||
|
|
header = f.read()
|
||
|
|
with open(os.path.join(base_dir, 'pythonsupport.h')) as f:
|
||
|
|
header += f.read()
|
||
|
|
|
||
|
|
def check_name(name: str) -> None:
|
||
|
|
if name.startswith('CPy'):
|
||
|
|
assert re.search(fr'\b{name}\b', header), (
|
||
|
|
f'"{name}" is used in mypyc.primitives but not declared in CPy.h')
|
||
|
|
|
||
|
|
for values in [registry.method_call_ops.values(),
|
||
|
|
registry.function_ops.values(),
|
||
|
|
registry.binary_ops.values(),
|
||
|
|
registry.unary_ops.values()]:
|
||
|
|
for ops in values:
|
||
|
|
if isinstance(ops, CFunctionDescription):
|
||
|
|
ops = [ops]
|
||
|
|
for op in ops:
|
||
|
|
check_name(op.c_function_name)
|
||
|
|
|
||
|
|
primitives_path = os.path.join(os.path.dirname(__file__), '..', 'primitives')
|
||
|
|
for fnam in glob.glob(f'{primitives_path}/*.py'):
|
||
|
|
with open(fnam) as f:
|
||
|
|
content = f.read()
|
||
|
|
for name in re.findall(r'c_function_name=["\'](CPy[A-Z_a-z0-9]+)', content):
|
||
|
|
check_name(name)
|