-
Notifications
You must be signed in to change notification settings - Fork 7
/
Copy pathRecommenderServiceTest.php
executable file
·94 lines (78 loc) · 3.09 KB
/
RecommenderServiceTest.php
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
<?php
use OpenCF\Algorithms\Similarity\Cosine;
use OpenCF\Algorithms\Similarity\WeightedCosine;
use OpenCF\Algorithms\Slopeone\WeightedSlopeone;
use OpenCF\Contracts\IRecommender;
use OpenCF\Exceptions\EmptyDatasetException;
use OpenCF\Exceptions\NotRegisteredRecommenderException;
use OpenCF\RecommenderService;
it('does not initialize with an empty dataset', function () {
new RecommenderService([]);
})->throws(EmptyDatasetException::class);
it('does not register an unsupported recommender', function () {
$dataset = [
'item1' => ['rating1' => 4],
];
$recommenderService = new RecommenderService($dataset);
$recommenderService->getRecommender('Cosine');
})->throws(NotRegisteredRecommenderException::class);
it('registers a supported recommender', function () {
$dataset = [
'item1' => ['rating1' => 4],
];
$recommenderService = new RecommenderService($dataset);
// instance of cosine
$class = new class () implements IRecommender {
public function name(): string
{
return 'Test Recommender';
}
public function getModel(): array
{
return [];
}
public function buildModel(): IRecommender
{
return $this;
}
public function predict(array $userRatings): array
{
return [];
}
};
$recommenderService->registerRecommender(get_class($class));
// instance of weighted cosine
$recommenderService->registerRecommender(WeightedCosine::class);
expect($recommenderService->getRecommender(WeightedCosine::class))->toBeInstanceOf(WeightedCosine::class);
// instance of weighted cosine
$recommenderService->registerRecommender(WeightedSlopeone::class);
expect($recommenderService->getRecommender(WeightedSlopeone::class))->toBeInstanceOf(WeightedSlopeone::class);
});
it('throws an exception when retrieving an invalid recommender', function () {
$dataset = [
'item1' => ['rating1' => 4],
];
$recommenderService = new RecommenderService($dataset);
$recommenderService->getRecommender('Invalid');
})->throws(NotRegisteredRecommenderException::class);
it('retrieves a registered recommender', function () {
$dataset = [
'item1' => ['rating1' => 4],
];
$recommenderService = new RecommenderService($dataset);
$recommenderService->registerRecommender(Cosine::class);
$instance = $recommenderService->getRecommender(Cosine::class);
expect($instance)->toBeInstanceOf(Cosine::class);
});
it('retrieves a registers all default recommenders', function () {
$dataset = [
'item1' => ['rating1' => 4],
];
$recommenderService = new RecommenderService($dataset);
$instance = $recommenderService->getRecommender(Cosine::class);
expect($instance)->toBeInstanceOf(Cosine::class);
$instance = $recommenderService->getRecommender(WeightedCosine::class);
expect($instance)->toBeInstanceOf(WeightedCosine::class);
$instance = $recommenderService->getRecommender(WeightedSlopeone::class);
expect($instance)->toBeInstanceOf(WeightedSlopeone::class);
});