diff options
Diffstat (limited to 'RhSolutions.ML.Tests/Tests.cs')
| -rw-r--r-- | RhSolutions.ML.Tests/Tests.cs | 13 |
1 files changed, 13 insertions, 0 deletions
diff --git a/RhSolutions.ML.Tests/Tests.cs b/RhSolutions.ML.Tests/Tests.cs index 77f6991..e7329ce 100644 --- a/RhSolutions.ML.Tests/Tests.cs +++ b/RhSolutions.ML.Tests/Tests.cs @@ -117,5 +117,18 @@ public class Tests }; var prediction = _predEngine.Predict(p); Assert.That(prediction.Type, Is.EqualTo("Pink")); + } + + [TestCase("Труба канализационная")] + [TestCase("Труба 110")] + [TestCase("Труба 110 500")] + public void WastePipeTest(string name) + { + Product p = new() + { + Name = name + }; + var prediction = _predEngine.Predict(p); + Assert.That(prediction.Type, Is.EqualTo("Труба RAUPIANO")); } }
\ No newline at end of file |