diff options
Diffstat (limited to 'RhSolutions.ML.Tests')
-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 920c666..e38e459 100644 --- a/RhSolutions.ML.Tests/Tests.cs +++ b/RhSolutions.ML.Tests/Tests.cs @@ -94,4 +94,16 @@ public class Tests 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")); + } }
\ No newline at end of file |