diff options
Diffstat (limited to 'RhSolutions.ML.Tests/Tests.cs')
| -rw-r--r-- | RhSolutions.ML.Tests/Tests.cs | 12 |
1 files changed, 12 insertions, 0 deletions
diff --git a/RhSolutions.ML.Tests/Tests.cs b/RhSolutions.ML.Tests/Tests.cs index e38e459..77f6991 100644 --- a/RhSolutions.ML.Tests/Tests.cs +++ b/RhSolutions.ML.Tests/Tests.cs @@ -106,4 +106,16 @@ public class Tests 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")); + } }
\ No newline at end of file |