diff --git a/test.py b/test.py index 32c5d97..abc3fc9 100644 --- a/test.py +++ b/test.py @@ -1,9 +1,12 @@ import importlib +from pathlib import Path from lexer.annotations import AnnotationLexer +from lexer.midas import MidasLexer from lexer.token import Token +# Frame annotation mod = importlib.import_module("examples.00_syntax_prototype.01_simple_types") annotation: str = mod.__annotations__["df"] @@ -12,4 +15,14 @@ tokens: list[Token] = lexer.process() print([ f"{t.type.name}('{t.lexeme}')" for t in tokens -]) \ No newline at end of file +]) + +# Midas type definitions +path: Path = Path("examples") / "00_syntax_prototype" / "02_custom_types.midas" +definitions: str = path.read_text() +midas_lexer: MidasLexer = MidasLexer(definitions, path.name) +tokens = midas_lexer.process() +print([ + f"{t.type.name}('{t.lexeme}')" + for t in tokens +])