AssetRipper_AssetRipper/Source/AssetRipper.Tests/PackedBitVectorTests/FloatVectorTests.cs

using AssetRipper.SourceGenerated.Extensions;
using AssetRipper.SourceGenerated.Subclasses.PackedBitVector_Single;

namespace AssetRipper.Tests.PackedBitVectorTests;

/// <summary>
/// Tests for <see cref="PackedBitVector_Single"/>
/// </summary>
public class FloatVectorTests
{
	private static readonly Random random = new Random(57089);
	private static readonly float[] floats = MakeFloats(20);

	private static float[] MakeFloats(int count)
	{
		float[] result = new float[count];
		for (int i = 0; i < count; i++)
		{
			result[i] = 10f * random.NextSingle();
		}
		return result;
	}

	[Test]
	public void BitSize24ProducesHighlyAccurateResults()
	{
		PackedBitVector_Single packedVector = new();
		packedVector.Pack(floats, 24, false);
		float[] unpackedValues = packedVector.Unpack();
		Assert.That(unpackedValues, Is.EqualTo(floats).Within(0.000001f));
	}

	[Test]
	public void BitSizeAdjustmentAlsoProducesUsableResults()
	{
		PackedBitVector_Single packedVector = new();
		packedVector.Pack(floats, 8, true);
		float[] unpackedValues = packedVector.Unpack();
		Assert.That(unpackedValues, Is.EqualTo(floats).Within(0.01f));
	}

	[Theory]
	public void PackingAndUnpackingDoesNotThrowAndGivesCorrectLength([Range(1, 32)] int bitSize)
	{
		PackedBitVector_Single packedVector = new();
		packedVector.Pack(floats, bitSize, false);
		float[] unpackedValues = packedVector.Unpack();
		Assert.That(unpackedValues, Has.Length.EqualTo(floats.Length));
	}

	[Test]
	public void DifferingChunkStrideAndChunkSizeWorks()
	{
		Assert.That(floats.Length % 4 == 0);
		PackedBitVector_Single packedVector = new();
		packedVector.Pack(floats);
		float[] actual = packedVector.Unpack(3, 4);
		float[] expected = new float[floats.Length / 4 * 3];
		for (int chunk = 0; chunk < floats.Length / 4; chunk++)
		{
			for (int i = 0; i < 3; i++)
			{
				expected[chunk * 3 + i] = floats[chunk * 4 + i];
			}
		}
		Assert.That(actual, Is.EqualTo(expected).Within(0.000001f));
	}
}