add private_tests_dir to make tester (#28)

manytask · Sep 5, 2023 · 17210d5 · 17210d5
1 parent cd1e73c
commit 17210d5
Showing 1 changed file with 18 additions and 7 deletions.
diff --git a/checker/testers/make.py b/checker/testers/make.py
@@ -15,6 +15,7 @@ class TaskTestConfig(Tester.TaskTestConfig):
         test_timeout: int = 60  # seconds
 
         public_test_files: list[str] = field(default_factory=list)
+        private_test_files: list[str] = field(default_factory=list)
 
     def _gen_build(  # type: ignore[override]
             self,
@@ -35,13 +36,23 @@ def _gen_build(  # type: ignore[override]
             verbose=verbose,
         )
 
-        self._executor(
-            copy_files,
-            source=public_tests_dir,
-            target=build_dir,
-            patterns=test_config.public_test_files,
-            verbose=verbose,
-        )
+        if public_tests_dir is not None:
+            self._executor(
+                copy_files,
+                source=public_tests_dir,
+                target=build_dir,
+                patterns=test_config.public_test_files,
+                verbose=verbose,
+            )
+
+        if private_tests_dir is not None:
+            self._executor(
+                copy_files,
+                source=private_tests_dir,
+                target=build_dir,
+                patterns=test_config.private_test_files,
+                verbose=verbose,
+            )
 
     def _clean_build(  # type: ignore[override]
             self,