Revert "use old yamato test config" (#4233)

* Revert "use old yamato test config (#4231)"

This reverts commit e5e21dcc2ae9a226b1e4017fa1404abe54ef83a6.

* Apply changes from #4232

.yamato/com.unity.ml-agents-test.yml

     type: Unity::VM
     image: package-ci/ubuntu:stable
     flavor: b1.medium
+packages:
+  - name: com.unity.ml-agents
+  - name: com.unity.ml-agents.extensions
 ---
 
 all_package_tests:
     {% for platform in test_platforms %}
-  - .yamato/com.unity.ml-agents-test.yml#test_{{ platform.name }}_{{ editor.version }}
+  - .yamato/com.unity.ml-agents-test.yml#test_com.unity.ml-agents_{{ platform.name }}_{{ editor.version }}
-  - .yamato/com.unity.ml-agents-test.yml#test_{{ platform.name }}_{{ editor.version }}
+  - .yamato/com.unity.ml-agents-test.yml#test_com.unity.ml-agents_{{ platform.name }}_{{ editor.version }}
     {% endfor %}
   {% endfor %}
   triggers:
         frequency: daily
 
-{% for editor in test_editors %}
-  {% for platform in test_platforms %}
-test_{{ platform.name }}_{{ editor.version }}:
-  name : com.unity.ml-agents test {{ editor.version }} on {{ platform.name }}
+{% for package in packages %}
+  {% for editor in test_editors %}
+    {% for platform in test_platforms %}
+test_{{ package.name }}_{{ platform.name }}_{{ editor.version }}:
+  name : {{ package.name }} test {{ editor.version }} on {{ platform.name }}
   agent:
     type: {{ platform.type }}
     image: {{ platform.image }}
-    - upm-ci package test -u {{ editor.version }} --package-path com.unity.ml-agents {{ editor.coverageOptions }}
+    - upm-ci project test -u {{ editor.version }} --project-path Project --package-filter {{ package.name }} {{ editor.coverageOptions }}
+
+    {% if package.name == "com.unity.ml-agents" %}
+    # TODO get coverage tests running for extensions too
+    {% endif %}
   artifacts:
     logs:
       paths:
       pull_request.target match "release.+") AND
       NOT pull_request.draft AND
       (pull_request.changes.any match "com.unity.ml-agents/**" OR
+      {% if package.name == "com.unity.ml-agents.extensions" %}
+        pull_request.changes.any match "com.unity.ml-agents.extensions/**" OR
+      {% endif %}
+    {% endfor %}
-{% for editor in trunk_editor %}
-  {% for platform in test_platforms %}
-test_{{ platform.name }}_trunk:
-  name : com.unity.ml-agents test {{ editor.version }} on {{ platform.name }}
+{% for package in packages %}
+  {% for editor in trunk_editor %}
+    {% for platform in test_platforms %}
+test_{{ package.name }}_{{ platform.name }}_trunk:
+  name : {{ package.name }} test {{ editor.version }} on {{ platform.name }}
   agent:
     type: {{ platform.type }}
     image: {{ platform.image }}
     - unity-downloader-cli -u trunk -c editor --wait --fast
     - npm install upm-ci-utils@stable -g --registry https://artifactory.prd.cds.internal.unity3d.com/artifactory/api/npm/upm-npm
-    - upm-ci package test -u {{ editor.version }} --package-path com.unity.ml-agents {{ editor.coverageOptions }}
+    - upm-ci project test -u {{ editor.version }} --project-path Project --package-filter {{ package.name }} {{ editor.coverageOptions }}
+    {% if package.name == "com.unity.ml-agents" %}
+    # TODO get coverage tests running for extensions too
+    {% endif %}
   artifacts:
     logs:
       paths:
     cancel_old_ci: true
     {% endfor %}
   {% endfor %}
+{% endfor %}