diff options
Diffstat (limited to 'RhSolutions.ML.Tests/RhSolutionsTests.cs')
| -rw-r--r-- | RhSolutions.ML.Tests/RhSolutionsTests.cs | 27 |
1 files changed, 27 insertions, 0 deletions
diff --git a/RhSolutions.ML.Tests/RhSolutionsTests.cs b/RhSolutions.ML.Tests/RhSolutionsTests.cs new file mode 100644 index 0000000..e1ec8f4 --- /dev/null +++ b/RhSolutions.ML.Tests/RhSolutionsTests.cs @@ -0,0 +1,27 @@ +namespace RhSolutions.ML.Tests; + +public abstract class RhSolutionsTests +{ + 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; + + [SetUp] + public void Setup() + { + _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)); + } +}
\ No newline at end of file |