public void CreateMovieMatrix_TwoMovies_VectorizesMovieAttributesCorrectly()
{
var movies = CreateMovies();
var tags = CreateTags();
var creators = CreateCreators();
var vectorizer = new Service.MovieVectorizer(tags, creators);
var matrixString = JsonConvert.SerializeObject(vectorizer.CreateMovieMatrix(movies));
Assert.AreEqual("[[0.8,0.0,1.0,1.0,1.0,1.0,0.0],[0.7,0.0,1.0,0.0,1.0,0.0,1.0]]", matrixString);
}
public void CreateMovieMatrix_TwoMoviesWithCustomWeights_VectorizesMovieAttributesCorrectly()
{
var movies = CreateMovies();
var tags = CreateTags();
var creators = CreateCreators();
var weights = new Service.MovieVectorizer.Weights()
{
Creators = 10.0f,
Descriptions = 9.0f,
Rating = 8.0f,
Tags = 7.0f,
Year = 6.0f
};
var vectorizer = new Service.MovieVectorizer(tags, creators);
var matrixString = JsonConvert.SerializeObject(vectorizer.CreateMovieMatrix(movies, weights));
Assert.AreEqual("[[6.4,0.0,7.0,7.0,7.0,10.0,0.0],[5.6,0.0,7.0,0.0,7.0,0.0,10.0]]", matrixString);
}
