diff --git a/src/api/comp_method_train.yaml b/src/api/comp_method_train.yaml
index 043f7e4..4cb81d3 100644
--- a/src/api/comp_method_train.yaml
+++ b/src/api/comp_method_train.yaml
@@ -22,4 +22,11 @@ arguments:
   - name: "--output"
     __merge__: file_pretrained_model.yaml
     direction: output
-    required: true
\ No newline at end of file
+    required: true
+test_resources:
+  - type: python_script
+    path: /common/component_tests/check_config.py
+  - type: python_script
+    path: /common/component_tests/run_and_check_output.py
+  - path: /resources_test/predict_modality/openproblems_neurips2021/bmmc_cite/swap
+    dest: resources_test/predict_modality/openproblems_neurips2021/bmmc_cite/swap