Made a small fix

tested/nat_translation.py

@@ -120,12 +120,15 @@ def translate_contexts(contexts: list, language: str) -> list:
     result = []
     for context in contexts:
         assert isinstance(context, dict)
-        print(f"context: {context}")
-        if "script" in context or "testcases" in context:
-            key_to_set = "script" if "script" in context else "testcases"
-            raw_testcases = context.get(key_to_set)
-            assert isinstance(raw_testcases, list)
-            context[key_to_set] = translate_testcases(raw_testcases, language)
+        key_to_set = "script" if "script" in context else "testcases"
+        raw_testcases = context.get(key_to_set)
+        assert isinstance(raw_testcases, list)
+        context[key_to_set] = translate_testcases(raw_testcases, language)
+        if "files" in context:
+            files = context.get("files")
+            if isinstance(files, NaturalLanguageMap):
+                assert language in files
+                context["files"] = files[language]
         result.append(context)
 
     return result

tests/test_dsl_yaml.py

@@ -1360,11 +1360,13 @@ def test_natural_translate_unit_test():
         description: !natural_language
           en: "Ten"
           nl: "Tien"
-      files:
-        - name: "file.txt"
-          url: "media/workdir/file.txt"
-        - name: "fileNL.txt"
-          url: "media/workdir/fileNL.txt"
+      files: !natural_language
+        en:
+          - name: "file.txt"
+            url: "media/workdir/file.txt"
+        nl:
+          - name: "fileNL.txt"
+            url: "media/workdir/fileNL.txt"
     - testcases:
         - statement: !natural_language
             en: 'result = Trying(11)'
@@ -1400,8 +1402,6 @@ def test_natural_translate_unit_test():
     files:
     - name: file.txt
       url: media/workdir/file.txt
-    - name: fileNL.txt
-      url: media/workdir/fileNL.txt
   - testcases:
     - statement: result = Trying(11)
     - expression: result