fixed linting

tested/dsl/translate_parser.py

@@ -83,16 +83,27 @@ class TestedType:
 class ExpressionString(str):
     pass
 
+
 class ReturnOracle(dict):
     pass
 
+
 class NaturalLanguageMap(dict):
     pass
 
 
 OptionDict = dict[str, int | bool]
 YamlObject = (
-    YamlDict | list | bool | float | int | str | None | ExpressionString | ReturnOracle | NaturalLanguageMap
+    YamlDict
+    | list
+    | bool
+    | float
+    | int
+    | str
+    | None
+    | ExpressionString
+    | ReturnOracle
+    | NaturalLanguageMap
 )
 
 
@@ -139,13 +150,15 @@ def _return_oracle(loader: yaml.Loader, node: yaml.Node) -> ReturnOracle:
     ), f"A custom oracle must be an object, got {result} which is a {type(result)}."
     return ReturnOracle(result)
 
+
 def _natural_language_map(loader: yaml.Loader, node: yaml.Node) -> NaturalLanguageMap:
     result = _parse_yaml_value(loader, node)
     assert isinstance(
         result, dict
     ), f"A natural language map must be an object, got {result} which is a {type(result)}."
     return NaturalLanguageMap(result)
 
+
 def _parse_yaml(yaml_stream: str) -> YamlObject:
     """
     Parse a string or stream to YAML.

tested/nat_translation.py

@@ -2,8 +2,15 @@
 from typing import cast
 
 import yaml
-from tested.dsl.translate_parser import _parse_yaml, YamlObject, YamlDict, ReturnOracle, \
-    ExpressionString, _validate_testcase_combinations, NaturalLanguageMap
+from tested.dsl.translate_parser import (
+    _parse_yaml,
+    YamlObject,
+    YamlDict,
+    ReturnOracle,
+    ExpressionString,
+    _validate_testcase_combinations,
+    NaturalLanguageMap,
+)
 
 
 def translate_testcase(testcase: YamlDict, language: str) -> YamlDict:
@@ -79,13 +86,16 @@ def translate_testcase(testcase: YamlDict, language: str) -> YamlDict:
         if isinstance(description, NaturalLanguageMap):
             assert language in description
             testcase["description"] = description[language]
-        elif isinstance(description, dict) and isinstance(description["description"], dict):
+        elif isinstance(description, dict) and isinstance(
+            description["description"], dict
+        ):
             assert language in description["description"]
             description["description"] = description["description"][language]
             testcase["description"] = description
 
     return testcase
 
+
 def translate_testcases(testcases: list, language: str) -> list:
     result = []
     for testcase in testcases:
@@ -94,6 +104,7 @@ def translate_testcases(testcases: list, language: str) -> list:
 
     return result
 
+
 def translate_contexts(contexts: list, language: str) -> list:
     result = []
     for context in contexts:
@@ -104,6 +115,7 @@ def translate_contexts(contexts: list, language: str) -> list:
 
     return result
 
+
 def translate_tab(tab: YamlDict, language: str) -> YamlDict:
     key_to_set = "units" if "units" in tab else "tab"
     name = tab.get(key_to_set)
@@ -133,6 +145,7 @@ def translate_tab(tab: YamlDict, language: str) -> YamlDict:
         tab["scripts"] = translate_testcases(tab["scripts"], language)
     return tab
 
+
 def translate_tabs(dsl_list: list, language: str) -> list:
     result = []
     for tab in dsl_list:
@@ -141,7 +154,8 @@ def translate_tabs(dsl_list: list, language: str) -> list:
 
     return result
 
-def translate_dsl(dsl_object: YamlObject, language:str) -> YamlObject:
+
+def translate_dsl(dsl_object: YamlObject, language: str) -> YamlObject:
     if isinstance(dsl_object, list):
         return translate_tabs(dsl_object, language)
     else:
@@ -152,25 +166,28 @@ def translate_dsl(dsl_object: YamlObject, language:str) -> YamlObject:
         dsl_object[key_to_set] = translate_tabs(tab_list, language)
         return dsl_object
 
-def parse_yaml(yaml_path:str) -> YamlObject:
-    with open(yaml_path, 'r') as stream:
+
+def parse_yaml(yaml_path: str) -> YamlObject:
+    with open(yaml_path, "r") as stream:
         result = _parse_yaml(stream.read())
 
     return result
 
+
 def convert_to_yaml(yaml_object: YamlObject) -> str:
     def oracle_representer(dumper, data):
-        return dumper.represent_mapping('!oracle', data)
+        return dumper.represent_mapping("!oracle", data)
 
     def expression_representer(dumper, data):
-        return dumper.represent_scalar('!expression', data)
+        return dumper.represent_scalar("!expression", data)
 
     # Register the representer for the ReturnOracle object
     yaml.add_representer(ReturnOracle, oracle_representer)
     yaml.add_representer(ExpressionString, expression_representer)
     return yaml.dump(yaml_object, sort_keys=False)
 
-if __name__ == '__main__':
+
+if __name__ == "__main__":
     n = len(sys.argv)
     assert n > 1, "Expected atleast two argument (path to yaml file and language)."
 
@@ -179,4 +196,3 @@ def expression_representer(dumper, data):
     new_yaml = parse_yaml(path)
     translated_dsl = translate_dsl(new_yaml, lang)
     print(convert_to_yaml(translated_dsl))
-