Move test config to contrib

hplt-project · Feb 8, 2024 · a8e78f4 · a8e78f4
1 parent 9ec77d3
commit a8e78f4
Show file tree

Hide file tree

Showing 2 changed files with 9 additions and 6 deletions.
diff --git a/tests/test_config.yml → contrib/test_full_config.yml b/tests/test_config.yml → contrib/test_full_config.yml
@@ -30,10 +30,6 @@ end:
 modifiers:
   - UpperCase: 0.05
   - TitleCase: 0.05
-  - Tags: 0.08
-    custom_detok_src: null
-    custom_detok_trg: zh
-    template: "__source__ {src} __target__ {trg} __done__"
   - Prefix: 0.05
     min_words: 2
     max_words: 5
@@ -55,7 +51,10 @@ modifiers:
     random_space:  0.1 # Adds a random space in the string.
     repeated_char: 0.1 # Repeats a random word character.
     unichar:       0.1 # Replaces a random consecutive repeated letter with a single letter.
-
+  - Tags: 0.08
+    custom_detok_src: null
+    custom_detok_trg: zh
+    template: "__source__ {src} __target__ {trg} __done__"
 
 seed: 1111
 trainer: /usr/bin/cat
diff --git a/tests/test_trainer_cli.py b/tests/test_trainer_cli.py
@@ -35,7 +35,11 @@ def test_marian_log_args(self):
 		self.assertEqual({**vars(parsed), **expected}, vars(parsed))
 
 	def test_config_loader(self):
-		config_path = os.path.join(os.path.dirname(os.path.abspath(__file__)), 'test_config.yml')
+		config_path = os.path.join(
+			os.path.dirname(os.path.abspath(__file__)),
+			os.pardir,
+			'contrib',
+			'test_full_config.yml')
 		with open(config_path, 'r', encoding='utf-8') as fh:
 			config = yaml.safe_load(fh)