aswi2021tri-musketyri-gitlab/Server/ServerAppFunctionalTests/Predictor/PredictionControllerTests.cs @ 9a529f10

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 ServerApp.WeatherPredictionParser;
using ServerApp.Parser.Parsers;
using ServerApp.Parser.InputData;
using ServerApp.Connection.XMLProtocolHandler;
using ServerApp.DataDownload;
using log4net.Config;

namespace ServerApp.Predictor.Tests
{
    [TestClass()]
    public class PredictionControllerTests
    {
        static PredictionController instance;

        [AssemblyInitialize()]
        public static void ClassInit(TestContext context)
        {
            // setup logging service
            XmlConfigurator.Configure();
            System.IO.Directory.CreateDirectory(@".\data");
            DataDownloader dd = new DataDownloader(@".\data\dd\", "http://openstore.zcu.cz/", "OD_ZCU_{type}_{month}_{year}_{format}.zip", "http://wttr.in/Plzen,czechia?format=j1");
            instance = new PredictionController(new JsonParser(dd, new CsvDataLoader()), new DataParser(dd));

            dd.DownloadData(DataType.JIS, DataFormat.CSV, new DataDownload.Date(1, 2019), new DataDownload.Date(12, 2020));
            dd.DownloadData(DataType.STROJE, DataFormat.CSV, new DataDownload.Date(1, 2017), new DataDownload.Date(12, 2020));
            dd.DownloadData(DataType.POCASI, DataFormat.CSV, new DataDownload.Date(1, 2017), new DataDownload.Date(12, 2020));
            instance.Train();
        }

        [TestMethod()]
        public void PredictSingleTimeWeatherTest()
        {
            Request request = new Request()
            {
                useEndDate = false,
                useWeather = false,
                start = new Connection.XMLProtocolHandler.Date()
                {
                    day = DateTime.Now.Day,
                    month = DateTime.Now.Month,
                    year = DateTime.Now.Year,
                    hour = 10
                }
            };
            Response actual = instance.Predict(request);
            CollectionAssert.AllItemsAreInstancesOfType(actual.predicitons, typeof(Prediction));
            Assert.AreEqual(1, actual.predicitons.Length);
        }

        [TestMethod()]
        public void PredictStartEndWithinSingleDayTest()
        {
            Request request = new Request
            {
                useWeather = false,
                temperature = -0,
                wind = -0,
                rain = -0,
                weather = WeatherCondition.Sunny,
                useEndDate = true,
                start = new Connection.XMLProtocolHandler.Date
                {
                    day = 11,
                    month = 6,
                    year = 2021,
                    hour = 7
                },
                end = new Connection.XMLProtocolHandler.Date
                {
                    day = 11,
                    month = 6,
                    year = 2021,
                    hour = 12
                }
            };
            Response actual = instance.Predict(request);
            Assert.IsTrue(Math.Abs(actual.predicitons[0].dateTime.hour - request.start.hour) <= instance.Configuration.TimeResolution);
            Assert.IsTrue(Math.Abs(actual.predicitons[actual.predicitons.Length - 1].dateTime.hour - request.end.hour) <= instance.Configuration.TimeResolution);
        }

        [TestMethod()]
        public void PredictStartEndEqualTest()
        {
            Request request = new Request
            {
                useWeather = false,
                temperature = -0,
                wind = -0,
                rain = -0,
                weather = WeatherCondition.Sunny,
                useEndDate = true,
                start = new Connection.XMLProtocolHandler.Date
                {
                    day = 11,
                    month = 6,
                    year = 2021,
                    hour = 7
                },
                end = new Connection.XMLProtocolHandler.Date
                {
                    day = 11,
                    month = 6,
                    year = 2021,
                    hour = 7
                }
            };
            Response actual = instance.Predict(request);
            Assert.IsTrue(actual.predicitons.Length == 1);
        }

        [TestMethod()]
        public void RollbackTest()
        {
            instance.Train();
            int actual = instance.Rollback();
            Assert.AreEqual(0, actual);
        }
    }
}