| | | 1 | | using Is = NUnit.Framework.Is; |
| | | 2 | | using NUnit.Framework; |
| | | 3 | | using NUnit.Framework.Constraints; |
| | | 4 | | using System.Collections.Generic; |
| | | 5 | | using UnityEngine; |
| | | 6 | | using UnityEngine.TestTools; |
| | | 7 | | |
| | | 8 | | public class NearestNeighborInterpolator2DTest { |
| | | 9 | | // Helper method to count how many data points the 2D interpolator is holding. |
| | 1 | 10 | | private static int GetNumDataPoints(NearestNeighborInterpolator2D interpolator) { |
| | 1 | 11 | | var fieldInfo = typeof(NearestNeighborInterpolator2D) |
| | | 12 | | .GetField("_data", System.Reflection.BindingFlags.NonPublic | |
| | | 13 | | System.Reflection.BindingFlags.Instance); |
| | | 14 | | |
| | | 15 | | // Get the raw value and use IList.Count. |
| | 1 | 16 | | var dataList = fieldInfo.GetValue(interpolator) as System.Collections.IList; |
| | 1 | 17 | | return dataList?.Count ?? 0; |
| | 1 | 18 | | } |
| | | 19 | | |
| | | 20 | | [Test] |
| | 1 | 21 | | public void TestEmptyAndInvalidLines() { |
| | 1 | 22 | | string[] csvLines = { |
| | | 23 | | "", // Empty. |
| | | 24 | | " , ", // Whitespace. |
| | | 25 | | "abc,123,45,67" // Invalid because "abc" is not float. |
| | | 26 | | }; |
| | 1 | 27 | | NearestNeighborInterpolator2D interpolator = new NearestNeighborInterpolator2D(csvLines); |
| | | 28 | | |
| | | 29 | | // This should result in zero valid data points. |
| | 1 | 30 | | Assert.AreEqual(0, GetNumDataPoints(interpolator), "No valid lines should have been parsed."); |
| | | 31 | | |
| | | 32 | | // Expect the error log before calling the method that produces it. |
| | 1 | 33 | | LogAssert.Expect(LogType.Error, "No data points available for interpolation."); |
| | 1 | 34 | | Interpolator2DDataPoint result = interpolator.Interpolate(1, 1); |
| | 1 | 35 | | Assert.AreEqual(0, result.Data.Count, "Interpolator returned data from an empty dataset."); |
| | 1 | 36 | | } |
| | | 37 | | |
| | | 38 | | [Test] |
| | 1 | 39 | | public void TestSingleDataPoint() { |
| | 1 | 40 | | string[] csvLines = { "0,0,45,2" }; |
| | 1 | 41 | | NearestNeighborInterpolator2D interpolator = new NearestNeighborInterpolator2D(csvLines); |
| | | 42 | | |
| | | 43 | | // Query a random far point. |
| | 1 | 44 | | Interpolator2DDataPoint result = interpolator.Interpolate(100, 200); |
| | 1 | 45 | | Assert.AreEqual(new Vector2(0, 0), result.Coordinates); |
| | 1 | 46 | | Assert.AreEqual(2, result.Data.Count); |
| | 1 | 47 | | Assert.AreEqual(45f, result.Data[0]); |
| | 1 | 48 | | Assert.AreEqual(2f, result.Data[1]); |
| | 1 | 49 | | } |
| | | 50 | | |
| | | 51 | | [Test] |
| | 1 | 52 | | public void TestInsufficientColumns() { |
| | 1 | 53 | | string[] csvLines = { "100.0,200.0" }; |
| | 1 | 54 | | NearestNeighborInterpolator2D interpolator = new NearestNeighborInterpolator2D(csvLines); |
| | | 55 | | |
| | | 56 | | // Interpolate from the single data point. |
| | 1 | 57 | | Interpolator2DDataPoint result = interpolator.Interpolate(100.0f, 200.0f); |
| | 1 | 58 | | Assert.NotNull(result); |
| | 1 | 59 | | Assert.AreEqual(new Vector2(100.0f, 200.0f), result.Coordinates); |
| | 1 | 60 | | Assert.AreEqual(0, result.Data.Count, "Expected zero data columns for this line."); |
| | 1 | 61 | | } |
| | | 62 | | |
| | | 63 | | [Test] |
| | 1 | 64 | | public void TestOutOfRangeQuery() { |
| | | 65 | | // minimal data covering some small region. |
| | 1 | 66 | | string[] csvLines = { "0,0,45,2", "10,10,30,3", "20,5,25,4" }; |
| | 1 | 67 | | NearestNeighborInterpolator2D interpolator = new NearestNeighborInterpolator2D(csvLines); |
| | | 68 | | |
| | | 69 | | // Query something very far from all data points, e.g., (1000, 1000). |
| | 1 | 70 | | Interpolator2DDataPoint result = interpolator.Interpolate(1000, 1000); |
| | | 71 | | |
| | | 72 | | // The nearest neighbor is (20, 5), so we expect (25, 4). |
| | 1 | 73 | | Assert.AreEqual(new Vector2(20, 5), result.Coordinates); |
| | 1 | 74 | | Assert.AreEqual(2, result.Data.Count); |
| | 1 | 75 | | Assert.AreEqual(25f, result.Data[0]); |
| | 1 | 76 | | Assert.AreEqual(4f, result.Data[1]); |
| | 1 | 77 | | } |
| | | 78 | | |
| | | 79 | | [Test] |
| | 1 | 80 | | public void TestInterpolateVector2Success() { |
| | 1 | 81 | | string[] csvLines = { "1,1,10,20", "1,2,30,40", "2,1,50,60", "2,2,70,80" }; |
| | 1 | 82 | | NearestNeighborInterpolator2D interpolator = new NearestNeighborInterpolator2D(csvLines); |
| | | 83 | | |
| | 1 | 84 | | Vector2 testPoint = new Vector2(1.5f, 1.5f); |
| | 1 | 85 | | Interpolator2DDataPoint result = interpolator.Interpolate(testPoint); |
| | 1 | 86 | | Assert.AreEqual(2, result.Data.Count); |
| | | 87 | | |
| | | 88 | | // Check if the result matches any of the possible nearest point pairs. |
| | 1 | 89 | | bool isValidPair = (result.Data[0] == 10f && result.Data[1] == 20f) || |
| | | 90 | | (result.Data[0] == 30f && result.Data[1] == 40f) || |
| | | 91 | | (result.Data[0] == 50f && result.Data[1] == 60f) || |
| | | 92 | | (result.Data[0] == 70f && result.Data[1] == 80f); |
| | 1 | 93 | | Assert.IsTrue(isValidPair, "Result should match one of the nearest point pairs."); |
| | 1 | 94 | | } |
| | | 95 | | |
| | | 96 | | [Test] |
| | 1 | 97 | | public void TestInterpolateVector2Error() { |
| | | 98 | | // Empty dataset. |
| | 1 | 99 | | string[] csvLines = {}; |
| | 1 | 100 | | NearestNeighborInterpolator2D interpolator = new NearestNeighborInterpolator2D(csvLines); |
| | | 101 | | |
| | 1 | 102 | | Vector2 testPoint = new Vector2(1.0f, 1.0f); |
| | 1 | 103 | | LogAssert.Expect(LogType.Error, "No data points available for interpolation."); |
| | 1 | 104 | | Interpolator2DDataPoint result = interpolator.Interpolate(testPoint); |
| | 1 | 105 | | Assert.AreEqual(0, result.Data.Count, "Should return empty list when no data is available."); |
| | 1 | 106 | | } |
| | | 107 | | } |