Merge pull request #12 from AnnulusGames/add-tests

Add: WeightedListTests

src/RandomExtensions/AssemblyInfo.cs

@@ -0,0 +1,3 @@
+using System.Runtime.CompilerServices;
+
+[assembly: InternalsVisibleTo("RandomExtensions.Tests")]

src/RandomExtensions/Collections/WeightedList.cs

@@ -72,6 +72,8 @@ public WeightedList()
     readonly List<WeightedValue<T>> list;
     readonly ValueCollection values;
 
+    internal double TotalWeight => totalWeight;
+
     public WeightedValue<T> this[int index]
     {
         get => list[index];

tests/RandomExtensions.Tests/WeightedListTests.cs

@@ -5,6 +5,44 @@ namespace RandomExtensions.Tests;
 
 public class WeightedListTests
 {
+    [Test]
+    public void Test_Add()
+    {
+        var list = new WeightedList<int>();
+
+        list.Add(0, 1);
+        Assert.That(list.Count, Is.EqualTo(1));
+        Assert.That(list[0].Value, Is.EqualTo(0));
+        Assert.That(list.TotalWeight, Is.EqualTo(1));
+
+        list.Add(1, 2);
+        Assert.That(list.Count, Is.EqualTo(2));
+        Assert.That(list[1].Value, Is.EqualTo(1));
+        Assert.That(list.TotalWeight, Is.EqualTo(3));
+    }
+
+    [Test]
+    public void Test_Remove()
+    {
+        var list = new WeightedList<int>
+        {
+            { 0, 1 },
+            { 1, 1 },
+            { 2, 1 },
+            { 3, 1 },
+        };
+
+        Assert.That(list.TotalWeight, Is.EqualTo(4));
+        list.RemoveAt(0);
+        Assert.That(list.TotalWeight, Is.EqualTo(3));
+        list.RemoveAt(0);
+        Assert.That(list.TotalWeight, Is.EqualTo(2));
+        list.RemoveAt(0);
+        Assert.That(list.TotalWeight, Is.EqualTo(1));
+        list.RemoveAt(0);
+        Assert.That(list.TotalWeight, Is.EqualTo(0));
+    }
+
     [Test]
     public void Test_RemoveRandom()
     {