您最多选择25个主题
主题必须以中文或者字母或数字开头,可以包含连字符 (-),并且长度不得超过35个字符
26 行
872 B
26 行
872 B
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);
|
|
}
|
|
|
|
}
|
|
}
|