| | | 1 | | using NUnit.Framework; |
| | | 2 | | using System.Collections; |
| | | 3 | | using System.Collections.Generic; |
| | | 4 | | using UnityEngine; |
| | | 5 | | using UnityEngine.TestTools; |
| | | 6 | | using System; |
| | | 7 | | |
| | | 8 | | public class LaunchAngleCsvInterpolatorTest { |
| | | 9 | | [Test] |
| | | 10 | | public void TestDataPoint() { |
| | | 11 | | LaunchAngleInput input = new LaunchAngleInput(distance: 9074.84f, altitude: 97.7306f); |
| | | 12 | | LaunchAngleOutput expectedOutput = |
| | | 13 | | new LaunchAngleOutput(launchAngle: 22f, timeToPosition: 21.32f); |
| | | 14 | | |
| | | 15 | | LaunchAngleCsvInterpolator interpolator = new LaunchAngleCsvInterpolator(); |
| | | 16 | | Assert.AreEqual(expectedOutput, interpolator.Plan(input)); |
| | | 17 | | } |
| | | 18 | | |
| | | 19 | | [Test] |
| | | 20 | | public void TestNearestNeighbor() { |
| | | 21 | | LaunchAngleInput input = new LaunchAngleInput(distance: 9076f, altitude: 94f); |
| | | 22 | | LaunchAngleOutput expectedOutput = |
| | | 23 | | new LaunchAngleOutput(launchAngle: 22f, timeToPosition: 21.32f); |
| | | 24 | | |
| | | 25 | | LaunchAngleCsvInterpolator interpolator = new LaunchAngleCsvInterpolator(); |
| | | 26 | | Assert.AreEqual(expectedOutput, interpolator.Plan(input)); |
| | | 27 | | } |
| | | 28 | | |
| | | 29 | | [Test] |
| | | 30 | | public void TestFileNotFound() { |
| | | 31 | | // Inject a mock ConfigLoader that returns an empty string. |
| | | 32 | | LaunchAngleCsvInterpolator.ConfigLoaderDelegate mockLoader = (string path) => ""; |
| | | 33 | | |
| | | 34 | | LogAssert.Expect(LogType.Error, "Failed to load CSV file from Planning/nonexistent.csv."); |
| | | 35 | | Assert.Throws<InvalidOperationException>(() => { |
| | | 36 | | var interpolator = new LaunchAngleCsvInterpolator("Planning/nonexistent.csv", mockLoader); |
| | | 37 | | interpolator.Plan(new LaunchAngleInput()); |
| | | 38 | | }); |
| | | 39 | | } |
| | | 40 | | |
| | | 41 | | [Test] |
| | | 42 | | public void TestInvalidInterpolationData() { |
| | | 43 | | // Create a mock CSV with invalid data format. |
| | | 44 | | string mockCsv = "invalid,csv,data\n1,2,3"; |
| | | 45 | | LaunchAngleCsvInterpolator.ConfigLoaderDelegate mockLoader = (string path) => mockCsv; |
| | | 46 | | |
| | | 47 | | var interpolator = new LaunchAngleCsvInterpolator(null, mockLoader); |
| | | 48 | | var input = new LaunchAngleInput(distance: 1, altitude: 2); |
| | | 49 | | |
| | | 50 | | var ex = Assert.Throws<InvalidOperationException>(() => { interpolator.Plan(input); }); |
| | | 51 | | Assert.That(ex.Message, Is.EqualTo("Interpolator returned invalid data.")); |
| | | 52 | | } |
| | | 53 | | } |
| | | 54 | | |
| | | 55 | | public class LaunchAngleDataInterpolatorTest { |
| | | 56 | | private class DummyLaunchAngleDataInterpolator : LaunchAngleDataInterpolator { |
| | 0 | 57 | | public DummyLaunchAngleDataInterpolator() : base() {} |
| | | 58 | | |
| | | 59 | | // Generate the list of launch angle data points to interpolate. |
| | 0 | 60 | | protected override List<LaunchAngleDataPoint> GenerateData() { |
| | 0 | 61 | | return new List<LaunchAngleDataPoint> { |
| | | 62 | | new LaunchAngleDataPoint(new LaunchAngleInput(distance: 1, altitude: 100), |
| | | 63 | | new LaunchAngleOutput(launchAngle: 90, timeToPosition: 10)), |
| | | 64 | | new LaunchAngleDataPoint(new LaunchAngleInput(distance: 100, altitude: 1), |
| | | 65 | | new LaunchAngleOutput(launchAngle: 10, timeToPosition: 20)), |
| | | 66 | | }; |
| | 0 | 67 | | } |
| | | 68 | | } |
| | | 69 | | |
| | | 70 | | [Test] |
| | 0 | 71 | | public void TestDataPoint() { |
| | 0 | 72 | | LaunchAngleInput input = new LaunchAngleInput(distance: 1, altitude: 100); |
| | 0 | 73 | | LaunchAngleOutput expectedOutput = new LaunchAngleOutput(launchAngle: 90, timeToPosition: 10); |
| | | 74 | | |
| | 0 | 75 | | DummyLaunchAngleDataInterpolator interpolator = new DummyLaunchAngleDataInterpolator(); |
| | 0 | 76 | | Assert.AreEqual(expectedOutput, interpolator.Plan(input)); |
| | 0 | 77 | | } |
| | | 78 | | |
| | | 79 | | [Test] |
| | 0 | 80 | | public void TestNearestNeighbor() { |
| | 0 | 81 | | LaunchAngleInput input = new LaunchAngleInput(distance: 1, altitude: 200); |
| | 0 | 82 | | LaunchAngleOutput expectedOutput = new LaunchAngleOutput(launchAngle: 90, timeToPosition: 10); |
| | | 83 | | |
| | 0 | 84 | | DummyLaunchAngleDataInterpolator interpolator = new DummyLaunchAngleDataInterpolator(); |
| | 0 | 85 | | Assert.AreEqual(expectedOutput, interpolator.Plan(input)); |
| | 0 | 86 | | } |
| | | 87 | | } |