File size: 1,062 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
using NUnit.Framework;
using Unity.MLAgents.Sensors;

namespace Unity.MLAgents.Tests
{
    [TestFixture]
    public class CompressionSpecTests
    {
        [Test]
        public void TestIsTrivialMapping()
        {
            Assert.IsTrue(CompressionSpec.Default().IsTrivialMapping());

            var spec = new CompressionSpec(SensorCompressionType.PNG, null);
            Assert.AreEqual(spec.IsTrivialMapping(), true);

            spec = new CompressionSpec(SensorCompressionType.PNG, new[] { 0, 0, 0 });
            Assert.AreEqual(spec.IsTrivialMapping(), true);

            spec = new CompressionSpec(SensorCompressionType.PNG, new[] { 0, 1, 2, 3, 4 });
            Assert.AreEqual(spec.IsTrivialMapping(), true);

            spec = new CompressionSpec(SensorCompressionType.PNG, new[] { 1, 2, 3, 4, -1, -1 });
            Assert.AreEqual(spec.IsTrivialMapping(), false);

            spec = new CompressionSpec(SensorCompressionType.PNG, new[] { 0, 0, 0, 1, 1, 1 });
            Assert.AreEqual(spec.IsTrivialMapping(), false);
        }
    }
}