File size: 1,216 Bytes
05c9ac2 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 |
using System.Collections;
using NUnit.Framework;
using UnityEngine.TestTools;
using UnityEngine;
using UnityEngine.SceneManagement;
using Unity.MLAgents;
#if UNITY_EDITOR
using UnityEditor.SceneManagement;
#endif
namespace Tests
{
public class AcademyStepperTest
{
[SetUp]
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>
[UnityTest]
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);
}
}
}
}
|