File size: 871 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
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);
        }
    }
}