create first tests

Author: filipebraida

Project.toml

@@ -11,13 +11,13 @@ Random = "9a3f8284-a2c9-5f02-9a11-845980a1fd5c"
 Statistics = "10745b16-79ce-11e8-11f9-7d13ad32a3b2"
 
 [compat]
-julia = "^1"
-Persa = "^0.1"
+julia = "1"
+Persa = "0.2"
 ProgressMeter = "^1.1"
 
 [extras]
-DataFrames = "a93c6f00-e57d-5684-b7b6-d8193f3e46c0"
+DatasetsCF = "6bac03b7-bcd8-55bf-9a08-86b9b673b6cc"
 Test = "8dfed614-e22c-5e08-85e1-65c5234f0b40"
 
 [targets]
-test = ["Test", "DataFrames"]
+test = ["Test", "DatasetsCF"]

test/runtests.jl

@@ -1,22 +1,37 @@
 using Persa
 using Test
-using DataFrames
-
-function createdummydatasetone()
-    df = DataFrame()
-    df[!, :user] = [1, 1, 1, 1, 1, 1, 2, 2, 2, 2, 2, 2, 3, 3, 3, 3, 4, 4, 4, 4, 4, 5, 5, 5, 5, 5, 5, 6, 6, 6, 6, 6, 7, 7, 7]
-    df[!, :item] = [1, 2, 3, 4, 5, 6, 1, 2, 3, 4, 5, 6, 1, 2, 3, 4, 2, 3, 4, 5, 6, 1, 2, 3, 4, 5, 6, 1, 2, 4, 5, 6, 2, 4, 5]
-    df[!, :rating] = [2.5, 3.5, 3.0, 3.5, 2.5, 3.0, 3, 3.5, 1.5, 5, 3, 3.5, 2.5, 3.0, 3.5, 4.0, 3.5, 3.0, 4.0, 2.5, 4.5, 3.0, 4.0, 2.0, 3.0, 2.0, 3.0, 3.0, 4.0, 5.0, 3.5, 3.0, 4.5, 4.0, 1.0]
-    return df
-end
+using DatasetsCF
+
+using ModelBasedCF
+
+dataset = DatasetsCF.MovieLens()
 
-function createdummydatasettwo()
-    df = DataFrame()
-    df[!, :user] = [1, 1, 1, 1, 1, 1, 2, 2, 2, 2, 2, 2, 3, 3, 3, 3, 4, 4, 4, 4, 4, 5, 5, 5, 5, 5, 5, 6, 6, 6, 6, 6, 7, 7, 7]
-    df[!, :item] = [1, 2, 3, 4, 5, 6, 1, 2, 3, 4, 5, 6, 1, 2, 3, 4, 2, 3, 4, 5, 6, 1, 2, 3, 4, 5, 6, 1, 2, 4, 5, 6, 2, 4, 5]
-    df[!, :rating] = [2, 7, 3, 3, 8, 3, 3, 3, 1, 9, 3, 3, 2, 3, 10, 4, 3, 3, 4, 2, 4, 3, 4, 2, 3, 2, 3, 3, 4, 5, 9, 10, 7, 6, 8]
-    return df
+@testset "Random Model Tests" begin
+    model = ModelBasedCF.RandomModel(dataset)
+    Persa.train!(model, dataset)
+
+    @test !isnan(model[1,1])
 end
-####
 
-@test true
+@testset "Factorization Matrix Models Tests" begin
+    @testset "Baseline Tests" begin
+        model = ModelBasedCF.Baseline(dataset)
+        Persa.train!(model, dataset, max_epochs = 1)
+
+        @test !isnan(model[1,1])
+    end
+
+    @testset "RSVD Tests" begin
+        model = ModelBasedCF.RSVD(dataset, 1)
+        Persa.train!(model, dataset, max_epochs = 1)
+
+        @test !isnan(model[1,1])
+    end
+
+    @testset "IRSVD Tests" begin
+        model = ModelBasedCF.IRSVD(dataset, 1)
+        Persa.train!(model, dataset, max_epochs = 1)
+
+        @test !isnan(model[1,1])
+    end
+end