From 292f5eaab423b00a62d95344900c670f4775efef Mon Sep 17 00:00:00 2001
From: Serghei Cebotari <serghei@cebotari.ru>
Date: Wed, 20 Sep 2023 14:41:16 +0300
Subject: Add pink pipe

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

(limited to 'RhSolutions.ML.Tests')

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