Cleaned up code and added extra cases.

tested/nat_translation.py

@@ -122,7 +122,6 @@ def translate_contexts(contexts: list, language: str) -> list:
         assert isinstance(context, dict)
         key_to_set = "script" if "script" in context else "testcases"
         raw_testcases = context.get(key_to_set)
-        print(raw_testcases)
         assert isinstance(raw_testcases, list)
         context[key_to_set] = translate_testcases(raw_testcases, language)
         if "files" in context:
@@ -158,7 +157,6 @@ def translate_tab(tab: YamlDict, language: str) -> YamlDict:
         assert isinstance(tab["testcases"], list)
         tab["testcases"] = translate_testcases(tab["testcases"], language)
     else:
-        print(tab)
         assert "scripts" in tab
         assert isinstance(tab["scripts"], list)
         tab["scripts"] = translate_testcases(tab["scripts"], language)
@@ -213,7 +211,6 @@ def expression_representer(dumper, data):
     path = sys.argv[1]
     lang = sys.argv[2]
     new_yaml = parse_yaml(path)
-    print(new_yaml)
     translated_dsl = translate_dsl(new_yaml, lang)
     yaml_string = convert_to_yaml(translated_dsl)
     print(yaml_string)

tests/test_dsl_yaml.py

@@ -1378,6 +1378,13 @@ def test_natural_translate_unit_test():
               en: "Eleven"
               nl: "Elf"
             format: "code"
+- tab: 'test'
+  testcases:
+    - expression: !natural_language
+        en: "tests(11)"
+        nl: "testen(11)"
+      return: 11
+
 """.strip()
     translated_yaml_str = """
 - tab: counting
@@ -1409,6 +1416,10 @@ def test_natural_translate_unit_test():
       description:
         description: Eleven
         format: code
+- tab: test
+  testcases:
+  - expression: tests(11)
+    return: 11
 """.strip()
     parsed_yaml = _parse_yaml(yaml_str)
     translated_dsl = translate_dsl(parsed_yaml, "en")
@@ -1466,6 +1477,12 @@ def test_natural_translate_io_test():
               nl: "Ziet er niet goed uit"
             types:
               typescript: "ERROR"
+  - unit: "test"
+    scripts:
+      - expression: !natural_language
+            en: "tests(11)"
+            nl: "testen(11)"
+        return: 11
 """.strip()
     translated_yaml_str = """
 units:
@@ -1495,6 +1512,10 @@ def test_natural_translate_io_test():
         message: Does not look good
         types:
           typescript: ERROR
+- unit: test
+  scripts:
+  - expression: tests(11)
+    return: 11
 """.strip()
     parsed_yaml = _parse_yaml(yaml_str)
     translated_dsl = translate_dsl(parsed_yaml, "en")