Implement Data driven tests

1 files changed, 39 insertions, 0 deletions

diff --git a/RhSolutions.ML.Tests/TestBase.cs b/RhSolutions.ML.Tests/TestBase.cs
new file mode 100644
index 0000000..679d9d3
--- /dev/null
+++ b/RhSolutions.ML.Tests/TestBase.cs
@@ -0,0 +1,39 @@
+using RhSolutions.ML.Lib;
+
+namespace RhSolutions.ML.Tests;
+
+public abstract class TestBase
+{
+	protected static string _appPath = Path.GetDirectoryName(Environment.GetCommandLineArgs()[0]) ?? ".";
+	protected static string _dataPath = Path.Combine(_appPath, "..", "..", "..", "..", "Models", "model.zip");
+	protected MLContext _mlContext;
+	protected PredictionEngine<Product, TypePrediction> _predEngine;
+	
+	public TestBase()
+	{
+		RhSolutionsMLBuilder.RebuildModel();
+		_mlContext = new MLContext(seed: 0);
+		ITransformer loadedNodel = _mlContext.Model.Load(_dataPath, out var _);
+		_predEngine = _mlContext.Model.CreatePredictionEngine<Product, TypePrediction>(loadedNodel);
+	}
+
+	public void Execute(string name, string expectedGroup)
+	{
+		Product p = new()
+		{
+			Name = name
+		};
+		var prediction = _predEngine.Predict(p);
+		Assert.That(prediction.Type, Is.EqualTo(expectedGroup));
+	}
+	
+	public void Execute(Product expected)
+	{
+		Product actual = new()
+		{
+			Name = expected.Name
+		};
+		var prediction = _predEngine.Predict(actual);
+		Assert.That(prediction.Type, Is.EqualTo(expected.Type));
+	}
+}
\ No newline at end of file