|
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/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); |
|
} |
|
} |
|
} |
|
|