From e0edc3793ea8f6882fd5568c9055b1049ccfcf28 Mon Sep 17 00:00:00 2001
From: Serghei Cebotari <serghei@cebotari.ru>
Date: Wed, 20 Sep 2023 14:35:57 +0300
Subject: Add Flex Pipe

---
 RhSolutions.ML.Tests/Tests.cs | 12 ++++++++++++
 1 file changed, 12 insertions(+)

(limited to 'RhSolutions.ML.Tests/Tests.cs')

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
-- 
cgit v1.2.3