ppo-SnowballTarget / DevProject /Assets /ML-Agents /Scripts /Tests /Runtime /AcademyTest /AcademyStepperTest.cs
| using System.Collections; | |
| using NUnit.Framework; | |
| using UnityEngine.TestTools; | |
| using UnityEngine; | |
| using UnityEngine.SceneManagement; | |
| using Unity.MLAgents; | |
| using UnityEditor.SceneManagement; | |
| namespace Tests | |
| { | |
| public class AcademyStepperTest | |
| { | |
| [] | |
| public void Setup() | |
| { | |
| // We need register the communicator first before accessing the Academy. | |
| CommunicatorFactory.Register<ICommunicator>(RpcCommunicator.Create); | |
| Academy.Instance.Dispose(); | |
| SceneManager.LoadScene("ML-Agents/Scripts/Tests/Runtime/AcademyTest/AcademyStepperTestScene"); | |
| var academy = Academy.Instance; | |
| } | |
| /// <summary> | |
| /// Verify in each update, the Academy is only stepped once. | |
| /// </summary> | |
| [] | |
| public IEnumerator AcademyStepperCleanupPasses() | |
| { | |
| var academy = Academy.Instance; | |
| int initStepCount = academy.TotalStepCount; | |
| for (var i = 0; i < 5; i++) | |
| { | |
| yield return new WaitForFixedUpdate(); | |
| Assert.True(academy.TotalStepCount - initStepCount == i + 1); | |
| } | |
| } | |
| } | |
| } | |