public void When_DataIsCorrect_Should_ReturnCorrectSplit()
{
var data = new List <MlNetPreSplitPredictionInputRow>()
{
new MlNetPreSplitPredictionInputRow()
{
Date = new DateTime(2000, 1, 1)
},
new MlNetPreSplitPredictionInputRow()
{
Date = new DateTime(2000, 1, 2)
},
new MlNetPreSplitPredictionInputRow()
{
Date = new DateTime(2000, 2, 1)
},
new MlNetPreSplitPredictionInputRow()
{
Date = new DateTime(2000, 2, 2)
},
new MlNetPreSplitPredictionInputRow()
{
Date = new DateTime(2000, 2, 3)
}
};
var strategy = new LastMonthDataAsTestDataSplitStrategy(new MlNetPreSplitMappingProfile());
var result = strategy.SplitTrainingAndTestData(data);
Assert.AreEqual(3, result.testDataView.Preview().RowView.Length);
Assert.AreEqual(2, result.trainingDataView.Preview().RowView.Length);
}
public void When_InputDataIsEmpty_Should_ThrowNullReferenceException()
{
var data = new List <MlNetPreSplitPredictionInputRow>();
var strategy = new LastMonthDataAsTestDataSplitStrategy(new MlNetPreSplitMappingProfile());
strategy.SplitTrainingAndTestData(data);
}
public void When_AllDataIsInTheSameMonth_Should_ReturnEmptyTrainingData()
{
var data = new List <MlNetPreSplitPredictionInputRow>()
{
new MlNetPreSplitPredictionInputRow()
{
Date = new DateTime(2000, 1, 1)
},
new MlNetPreSplitPredictionInputRow()
{
Date = new DateTime(2000, 1, 2)
}
};
var strategy = new LastMonthDataAsTestDataSplitStrategy(new MlNetPreSplitMappingProfile());
var result = strategy.SplitTrainingAndTestData(data);
Assert.AreEqual(data.Count, result.testDataView.Preview().RowView.Length);
Assert.AreEqual(0, result.trainingDataView.Preview().RowView.Length);
}
public void When_InputDataIsNull_Should_ThrowNullReferenceException()
{
var strategy = new LastMonthDataAsTestDataSplitStrategy(new MlNetPreSplitMappingProfile());
strategy.SplitTrainingAndTestData(null);
}