121 lines
3.2 KiB
C#
121 lines
3.2 KiB
C#
namespace RhSolutions.ML.Tests;
|
||
|
||
public class Tests
|
||
{
|
||
private static string _appPath = Path.GetDirectoryName(Environment.GetCommandLineArgs()[0]) ?? ".";
|
||
private static string _dataPath = Path.Combine(_appPath, "..", "..", "..", "..", "Models", "model.zip");
|
||
private MLContext _mlContext;
|
||
private 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);
|
||
}
|
||
|
||
[TestCase("Гильза 16")]
|
||
[TestCase("Пресс-втулка")]
|
||
public void SleevesTest(string name)
|
||
{
|
||
Product p = new()
|
||
{
|
||
Name = name
|
||
};
|
||
var prediction = _predEngine.Predict(p);
|
||
Assert.That(prediction.Type, Is.EqualTo("Монтажная гильза"));
|
||
}
|
||
|
||
[TestCase("Тройник 20")]
|
||
[TestCase("Тройник 20-16-16")]
|
||
[TestCase("Тройник 20х20х20")]
|
||
[TestCase("Тройник 32*32*32")]
|
||
[TestCase("Тройник 50-50-32")]
|
||
public void TPieceTest(string name)
|
||
{
|
||
Product p = new()
|
||
{
|
||
Name = name
|
||
};
|
||
var prediction = _predEngine.Predict(p);
|
||
Assert.That(prediction.Type, Is.EqualTo("Тройник RAUTITAN"));
|
||
}
|
||
|
||
[TestCase("Тройник 50/50/45")]
|
||
[TestCase("Тройник 50/50/45°")]
|
||
public void WastePipeBranchTest(string name)
|
||
{
|
||
Product p = new()
|
||
{
|
||
Name = name
|
||
};
|
||
var prediction = _predEngine.Predict(p);
|
||
Assert.That(prediction.Type, Is.EqualTo("Тройник RAUPIANO"));
|
||
}
|
||
|
||
[TestCase("Муфта соединительная равнопроходная 16 PX")]
|
||
[TestCase("Муфта 16")]
|
||
[TestCase("Переход 20-16")]
|
||
[TestCase("Переходник 20-16")]
|
||
public void CouplingTest(string name)
|
||
{
|
||
Product p = new()
|
||
{
|
||
Name = name
|
||
};
|
||
var prediction = _predEngine.Predict(p);
|
||
Assert.That(prediction.Type, Is.EqualTo("Муфта соединительная"));
|
||
}
|
||
|
||
[TestCase("Переходник с наружной резьбой 20-R 3/4 RX+")]
|
||
[TestCase("Переходник 16 1/2 НР")]
|
||
[TestCase("ПНР 16")]
|
||
[TestCase("Переход НР 16 1/2")]
|
||
[TestCase("Муфта НР 16 1/2")]
|
||
public void AdapterExternalTest(string name)
|
||
{
|
||
Product p = new()
|
||
{
|
||
Name = name
|
||
};
|
||
var prediction = _predEngine.Predict(p);
|
||
Assert.That(prediction.Type, Is.EqualTo("Переходник на наружную резьбу"));
|
||
}
|
||
|
||
[TestCase("Труба stabil 16")]
|
||
[TestCase("Труба stabil")]
|
||
public void StabilPipeTest(string name)
|
||
{
|
||
Product p = new()
|
||
{
|
||
Name = name
|
||
};
|
||
var prediction = _predEngine.Predict(p);
|
||
Assert.That(prediction.Type, Is.EqualTo("Stabil"));
|
||
}
|
||
|
||
[TestCase("Труба flex 16")]
|
||
[TestCase("Труба flex")]
|
||
public void FlexPipeTest(string name)
|
||
{
|
||
Product p = new()
|
||
{
|
||
Name = name
|
||
};
|
||
var prediction = _predEngine.Predict(p);
|
||
Assert.That(prediction.Type, Is.EqualTo("Flex"));
|
||
}
|
||
|
||
[TestCase("Труба pink 16")]
|
||
[TestCase("Труба pink")]
|
||
public void PinkPipeTest(string name)
|
||
{
|
||
Product p = new()
|
||
{
|
||
Name = name
|
||
};
|
||
var prediction = _predEngine.Predict(p);
|
||
Assert.That(prediction.Type, Is.EqualTo("Pink"));
|
||
}
|
||
} |