Automatic ruff lint fixes, plus some type additions and docs.

Author: ialarmedalien

tests/test_compiler/test_duckdb_compiler.py

@@ -1,7 +1,8 @@
 """
-Tests compilation of a specification to python
+Tests compilation of a specification to python.
 """
 
+import duckdb
 import pytest
 from linkml_runtime import SchemaView
 from linkml_runtime.dumpers import yaml_dumper
@@ -12,14 +13,16 @@
 
 
 @pytest.fixture
-def session():
+def session() -> Session:
+    """LinkML transformer session with schema and spec set."""
     session = Session()
     session.set_source_schema(SCHEMA1)
     session.set_transformer_specification(SPECIFICATION)
     return session
 
 
-def test_compile(session):
+def test_compile(session: Session) -> None:
+    """Test the DuckDb compiler."""
     compiler = SQLCompiler()
     assert session.transformer_specification is not None
     compiled = compiler.compile(session.transformer_specification)
@@ -37,8 +40,6 @@ def test_compile(session):
     print("Target DDL:")
     print(target_ddl)
 
-    import duckdb
-
     conn = duckdb.connect(":memory:")
     conn.execute(target_ddl)
     conn.execute(compiled.serialization)

tests/test_compiler/test_graphviz_compiler.py

@@ -1,5 +1,5 @@
 """
-Tests compilation of a specification to graphviz
+Tests compilation of a specification to graphviz.
 """
 
 import pytest
@@ -11,16 +11,17 @@
 
 
 @pytest.fixture
-def compiler():
+def compiler() -> GraphvizCompiler:
+    """Instantiate a GraphvizCompiler."""
     return GraphvizCompiler(
         source_schemaview=SchemaView(SCHEMA1),
     )
 
 
-def test_compile(compiler):
+def test_compile(compiler: GraphvizCompiler) -> None:
+    """Basic test of the graphviz compiler functionality."""
     spec = load_specification(SPECIFICATION)
     compiled = compiler.compile(spec)
-    # print(compiled.serialization)
     assert (
         "sourceMapping:id -> targetDenormMapping:subject_id [style=dashed]"
         in compiled.serialization

tests/test_compiler/test_markdown_compiler.py

@@ -1,5 +1,5 @@
 """
-Tests compilation of a specification to markdown
+Tests compilation of a specification to markdown.
 """
 
 import pytest
@@ -11,13 +11,15 @@
 
 
 @pytest.fixture
-def compiler():
+def compiler() -> MarkdownCompiler:
+    """Instantiate a MarkdownCompiler."""
     return MarkdownCompiler(
         source_schemaview=SchemaView(SCHEMA1),
     )
 
 
-def test_compile(compiler):
+def test_compile(compiler: MarkdownCompiler) -> None:
+    """Basic test of Markdown Compiler functioning."""
     assert compiler.template_name is not None
     spec = load_specification(SPECIFICATION)
     markdown = compiler.compile(spec).serialization

tests/test_compiler/test_python_compiler.py

@@ -1,5 +1,5 @@
 """
-Tests compilation of a specification to python
+Tests compilation of a specification to python.
 """
 
 import pytest
@@ -13,15 +13,17 @@
 
 
 @pytest.fixture
-def compiler():
+def compiler() -> PythonCompiler:
+    """Instantiate a Python Compiler."""
     return PythonCompiler(
         source_schemaview=SchemaView(SCHEMA1),
         source_python_module="tests.input.examples.personinfo_basic.model.personinfo_model",
         target_python_module="tests.input.examples.personinfo_basic.model.agent_model",
     )
 
 
-def test_compile(compiler):
+def test_compile(compiler: PythonCompiler) -> None:
+    """Basic test of Python Compiler functionality."""
     spec = load_specification(SPECIFICATION)
     pycode = compiler.compile(spec)
     # TODO: include imports so that code compiles