aswi2021tri-musketyri-gitlab/Server/ServerAppUnitTests/Predictor/NaiveBayesClassifierTests.cs @ master

using Microsoft.VisualStudio.TestTools.UnitTesting;
using ServerApp.Predictor;
//
// Author: Roman Kalivoda
//

using System;
using System.Collections.Generic;
using System.Linq;
using System.Text;
using System.Threading.Tasks;
using System.Text.RegularExpressions;

namespace ServerApp.Predictor.Tests
{
    [TestClass()]
    public class NaiveBayesClassifierTests
    {
        static IPredictor predictor;

        static readonly List<ModelInput> trainInput = new()
        {
                new ModelInput
                {
                    Label = "10%",
                    Temp = 12,
                    Hour = 7,
                    Rain = 10,
                    Time = DateTime.Now,
                    Wind = 10
                },
                new ModelInput
                {
                    Label = "100%",
                    Temp = 32,
                    Hour = 7,
                    Rain = 0,
                    Time = DateTime.Now,
                    Wind = 70
                },
                new ModelInput
                {
                    Label = "90%",
                    Temp = 32,
                    Hour = 15,
                    Rain = 20,
                    Time = DateTime.Now,
                    Wind = 50
                },
                new ModelInput
                {
                    Label = "70%",
                    Temp = 32,
                    Hour = 18,
                    Rain = 0,
                    Time = DateTime.Now,
                    Wind = 70
                },
                new ModelInput
                {
                    Label = "30%",
                    Temp = 6,
                    Hour = 10,
                    Rain = 70,
                    Time = DateTime.Now,
                    Wind = 30
                }
            };

        [ClassInitialize()]
        public static void SetUpClass(TestContext context)
        {
            predictor = new NaiveBayesClassifier();
            predictor.Fit(trainInput);
        }

        [TestMethod()]
        public void PredictTest()
        {
            string actual = predictor.Predict(new ModelInput
            {
                Temp = 19,
                Hour = 17,
                Rain = 20,
                Time = DateTime.Now,
                Wind = 7
            });
            StringAssert.Matches(actual, new Regex(@"\d0%"));
        }
    }
}