Add microgrid as default import to conftest.py

matthias-wende-frequenz · matthias-wende-frequenz · commit 3123a38d5cee · 2023-05-24T17:35:06.000+02:00
Simplifies examples when using high level microgrid sdk tools like
`microgrid.logical_meter()`.

Signed-off-by: Matthias Wende &lt;matthias.wende@frequenz.com&gt;
diff --git a/src/conftest.py b/src/conftest.py
@@ -122,6 +122,8 @@ def evaluate(self, example: Example) -> None | str:
         import_header.append(
             path_to_import_statement(Path(os.path.relpath(example.path)))
         )
+        # Add microgrid as default import
+        import_header.append("from frequenz.sdk import microgrid")
         imports_code = "\n".join(import_header)
 
         # Dedent the code example

Original file line number	Diff line number	Diff line change
`@@ -122,6 +122,8 @@ def evaluate(self, example: Example) -> None \| str:`
`122`	`122`	`import_header.append(`
`123`	`123`	`path_to_import_statement(Path(os.path.relpath(example.path)))`
`124`	`124`	`)`
	`125`	`+ # Add microgrid as default import`
	`126`	`+ import_header.append("from frequenz.sdk import microgrid")`
`125`	`127`	`imports_code = "\n".join(import_header)`
`126`	`128`
`127`	`129`	`# Dedent the code example`