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: [PATCH] Add Flex Pipe

---
 Data/rehau_flex.txt           | 15 +++++++++++++++
 RhSolutions.ML.Tests/Tests.cs | 12 ++++++++++++
 2 files changed, 27 insertions(+)
 create mode 100644 Data/rehau_flex.txt

diff --git a/Data/rehau_flex.txt b/Data/rehau_flex.txt
new file mode 100644
index 0000000..113258c
--- /dev/null
+++ b/Data/rehau_flex.txt
@@ -0,0 +1,15 @@
+"Унив труба RAUTITAN flex 16x2,2, бухта 100м"	Flex
+"Унив труба RAUTITAN flex 20x2,8, бухта 100м"	Flex
+"Унив труба RAUTITAN flex 25x3,5, бухта 50м"	Flex
+"Унив труба RAUTITAN flex 32x4,4, бухта 50м"	Flex
+"Унив труба RAUTITAN flex 16x2,2, прям отрезки 6м"	Flex
+"Унив труба RAUTITAN flex 20x2,8, прям отрезки 6м"	Flex
+"Унив труба RAUTITAN flex 25x3,5, прям отрезки 6м"	Flex
+"Унив труба RAUTITAN flex 32x4,4, прям отрезки 6м"	Flex
+"Унив труба RAUTITAN flex 40x5,5, прям отрезки 6м"	Flex
+"Унив труба RAUTITAN flex 50x6,9, прям отрезки 6м"	Flex
+"Унив труба RAUTITAN flex 63x8,6, прям отрезки 6м"	Flex
+"Унив труба РЕХАУ FLEX 16x2,2, бухта 100м"	Flex
+"Унив труба РЕХАУ FLEX 20x2,8, бухта 100м"	Flex
+"Унив труба РЕХАУ FLEX 25x3,5, бухта 50м"	Flex
+"Унив труба РЕХАУ FLEX 32x4,4, бухта 50м"	Flex
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