public void ShouldReturnFeaturesWithLabelFromCsv() { const string input = "6.1,2.6,5.6,1.4,virginica\n5.0,3.6,1.4,0.2,setosa\n5.5,2.3,4.0,1.3,versicolor"; var expected = new List <FeaturesWithLabel> { new FeaturesWithLabel(new List <double> { 6.1, 2.6, 5.6, 1.4 }, "virginica"), new FeaturesWithLabel(new List <double> { 5.0, 3.6, 1.4, 0.2 }, "setosa"), new FeaturesWithLabel(new List <double> { 5.5, 2.3, 4.0, 1.3 }, "versicolor") }; var result = CsvParser.CsvToFeaturesWithLabel(input).ToList(); Assert.AreEqual(expected.Count, result.Count); expected.Zip(result).ForEach(AssertFeaturesWithLabelEqual); }