Fixing type annotation for python3.6 compatibility

3 files changed, 14 insertions, 11 deletions

diff --git a/src/test/captured.py b/src/test/captured.py
index 9488149..2c33b78 100644
--- a/src/test/captured.py
+++ b/src/test/captured.py
@@ -6,15 +6,17 @@
 #    By: charles <me@cacharle.xyz>                  +#+  +:+       +#+         #
 #                                                 +#+#+#+#+#+   +#+            #
 #    Created: 2020/09/11 12:16:25 by charles           #+#    #+#              #
-#    Updated: 2021/01/31 02:01:30 by charles          ###   ########.fr        #
+#    Updated: 2021/01/31 03:29:22 by charles          ###   ########.fr        #
 #                                                                              #
 # ############################################################################ #
 
+from typing import List
+
 import config
 
 
 class Captured:
-    def __init__(self, output: str, status: int, files_content: list[str], is_timeout: bool = False):
+    def __init__(self, output: str, status: int, files_content: List[str], is_timeout: bool = False):
         """Captured class
            output:        captured content
            status:        command status
diff --git a/src/test/result.py b/src/test/result.py
index 3cc268d..4d060fa 100644
--- a/src/test/result.py
+++ b/src/test/result.py
@@ -6,18 +6,19 @@
 #    By: charles <me@cacharle.xyz>                  +#+  +:+       +#+         #
 #                                                 +#+#+#+#+#+   +#+            #
 #    Created: 2020/09/11 12:17:34 by charles           #+#    #+#              #
-#    Updated: 2021/01/31 02:28:58 by charles          ###   ########.fr        #
+#    Updated: 2021/01/31 03:35:53 by charles          ###   ########.fr        #
 #                                                                              #
 # ############################################################################ #
 
 import sys
 import re
-from typing import Match
+from typing import Match, List
 
 import config
 from test.captured import Captured
 
 
+# TODO split into BaseResult, Result and LeakResult
 class Result:
     RED_CHARS   = "\033[31m"
     GREEN_CHARS = "\033[32m"
@@ -28,7 +29,7 @@ class Result:
     def __init__(
         self,
         cmd: str,
-        file_names: list[str],
+        file_names: List[str],
         expected: Captured,
         actual: Captured,
         leak_output: str = None
diff --git a/src/test/test.py b/src/test/test.py
index 527b435..517e610 100644
--- a/src/test/test.py
+++ b/src/test/test.py
@@ -6,14 +6,14 @@
 #    By: charles <charles.cabergs@gmail.com>        +#+  +:+       +#+         #
 #                                                 +#+#+#+#+#+   +#+            #
 #    Created: 2020/06/16 21:48:50 by charles           #+#    #+#              #
-#    Updated: 2021/01/31 02:21:17 by charles          ###   ########.fr        #
+#    Updated: 2021/01/31 03:30:15 by charles          ###   ########.fr        #
 #                                                                              #
 # ############################################################################ #
 
 import os
 import sys
 import subprocess
-from typing import Optional
+from typing import Optional, List, Dict
 
 import config
 from test.captured import Captured
@@ -25,8 +25,8 @@ class Test:
     def __init__(self,
                  cmd: str,
                  setup: str = "",
-                 files: list[str] = [],
-                 exports: dict[str, str] = {},
+                 files: List[str] = [],
+                 exports: Dict[str, str] = {},
                  timeout: float = config.TIMEOUT,
                  signal=None,
                  hook=[],
@@ -73,7 +73,7 @@ class Test:
         self.result = Result(self.full_cmd, self.files, expected, actual)
         self.result.put(index)
 
-    def _run_sandboxed(self, shell_cmd: list[str]) -> Captured:
+    def _run_sandboxed(self, shell_cmd: List[str]) -> Captured:
         """ Run the command in a sandbox environment """
         with sandbox.context():
             if self.setup != "":
@@ -95,7 +95,7 @@ class Test:
                     sys.exit(1)
             return self._run_capture(shell_cmd)
 
-    def _run_capture(self, shell_cmd: list[str]) -> Captured:
+    def _run_capture(self, shell_cmd: List[str]) -> Captured:
         """ Capture the output (stdout and stderr)
             Capture the content of the watched files after the command is run
         """