public void PostMultiple_positiveText_True() { //Arrange MLContext mlContext = new MLContext(); DataViewSchema predictionPipelineSchema; PredictionEngine <SentimentData, SentimentPrediction> predictionEngine = mlContext .Model.CreatePredictionEngine <SentimentData, SentimentPrediction>(mlContext .Model.Load("C:/Users/iRebeca/facultate/.NET/.NetProject/TSense/Tests/MlModels/sentiment_model.zip", out predictionPipelineSchema)); SentimentController sentiment = new SentimentController(predictionEngine); SentimentData sentimentData = new SentimentData(); sentimentData.SentimentText = "It was great"; List <SentimentData> sentiments = new List <SentimentData>() { sentimentData }; //Act var result = (OkObjectResult)sentiment.PostMultiple(sentiments).Result; var prediction = (ML_Server.DataModels.Sentiment)result.Value; //Assert Assert.IsInstanceOfType(result.Value, typeof(ML_Server.DataModels.Sentiment)); Assert.AreEqual(true, prediction.Prediction, prediction.Probability.ToString()); }
public void Post_negativeText_False() { //Arrange MLContext mlContext = new MLContext(); DataViewSchema predictionPipelineSchema; ITransformer predictionPipeline = mlContext .Model.Load("C:/Users/iRebeca/facultate/.NET/.NetProject/TSense/Tests/MlModels/sentiment_model.zip", out predictionPipelineSchema); PredictionEngine <SentimentData, SentimentPrediction> predictionEngine = mlContext .Model.CreatePredictionEngine <SentimentData, SentimentPrediction>(predictionPipeline); SentimentController sentiment = new SentimentController(predictionEngine); SentimentData sentimentData = new SentimentData(); sentimentData.SentimentText = "It was bad"; //Act var result = (OkObjectResult)sentiment.Post(sentimentData).Result; var prediction = (SentimentPrediction)result.Value; //Assert Assert.IsInstanceOfType(result.Value, typeof(SentimentPrediction)); Assert.AreEqual(prediction.Prediction, false); }