| using NUnit.Framework; |
| using UnityEngine; |
| using UnityEditor; |
| using Unity.MLAgents.Policies; |
|
|
| namespace Unity.MLAgents.Tests |
| { |
| [TestFixture] |
| public class TestSerialization |
| { |
| const string k_oldPrefabPath = "Packages/com.unity.ml-agents.tests/Tests/Editor/TestModels/old_serialized_agent.prefab"; |
| const int k_numVecObs = 212; |
| const int k_numContinuousActions = 39; |
|
|
| [Test] |
| public void TestDeserialization() |
| { |
| var prefab = AssetDatabase.LoadAssetAtPath<GameObject>(k_oldPrefabPath); |
| var agent = GameObject.Instantiate(prefab); |
| var bp = agent.GetComponent<BehaviorParameters>(); |
| Assert.AreEqual(bp.BrainParameters.ActionSpec.NumContinuousActions, k_numContinuousActions); |
| Assert.AreEqual(bp.BrainParameters.VectorObservationSize, k_numVecObs); |
| } |
| } |
| } |
|
|