From f5d25ed7e8e56f70ea65d5e2d2b669b34e1d39ea Mon Sep 17 00:00:00 2001
From: Serghei Cebotari <serghei@cebotari.ru>
Date: Wed, 20 Sep 2023 14:51:18 +0300
Subject: [PATCH] Add black pipe

---
 Data/rehau_black.txt          |  6 ++++++
 RhSolutions.ML.Tests/Tests.cs | 12 ++++++++++++
 2 files changed, 18 insertions(+)
 create mode 100644 Data/rehau_black.txt

diff --git a/Data/rehau_black.txt b/Data/rehau_black.txt
new file mode 100644
index 0000000..73454e4
--- /dev/null
+++ b/Data/rehau_black.txt
@@ -0,0 +1,6 @@
+"Отоп  труба RAUTITAN black 16х2,2 мм, бухта 200 м"	Black
+"Отоп  труба RAUTITAN black 20х2,8 мм, бухта 180 м"	Black
+"Отоп  труба RAUTITAN black 25х3,5 мм, бухта 100 м"	Black
+"Отоп труба РЕХАУ BLACK 16х2,2 мм, бухта 200 м"	Black
+"Отоп труба РЕХАУ BLACK 20х2,8 мм, бухта 180 м"	Black
+"Отоп труба РЕХАУ BLACK 25х3,5 мм, бухта 100 м"	Black
diff --git a/RhSolutions.ML.Tests/Tests.cs b/RhSolutions.ML.Tests/Tests.cs
index e7329ce..0eb84a3 100644
--- a/RhSolutions.ML.Tests/Tests.cs
+++ b/RhSolutions.ML.Tests/Tests.cs
@@ -119,6 +119,18 @@ public class Tests
 		Assert.That(prediction.Type, Is.EqualTo("Pink"));		
 	}	
 	
+	[TestCase("Труба black 16")]
+	[TestCase("Труба black")]
+	public void BlackPipeTest(string name)
+	{
+		Product p = new()
+		{
+			Name = name
+		};
+		var prediction = _predEngine.Predict(p);
+		Assert.That(prediction.Type, Is.EqualTo("Black"));		
+	}	
+	
 	[TestCase("Труба канализационная")]
 	[TestCase("Труба 110")]	
 	[TestCase("Труба 110 500")]