add more matrix transformation tests

ayrat555 · May 13, 2017 · ed45ac8 · ed45ac8
1 parent 3615b3d
commit ed45ac8
Show file tree

Hide file tree

Showing 2 changed files with 55 additions and 16 deletions.
diff --git a/test/rock/links_test.exs b/test/rock/links_test.exs
@@ -47,21 +47,20 @@ defmodule Rock.LinksTest do
       |> Links.matrix(criterion)
 
     ^link_matrix = [
-       [0, 7, 7, 5, 5, 4, 5, 5, 4, 4, 5, 5, 2, 2],
-       [0, 0, 7, 5, 4, 5, 5, 4, 5, 4, 5, 5, 2, 2],
-       [0, 0, 0, 4, 5, 5, 4, 5, 5, 4, 5, 5, 2, 2],
-       [0, 0, 0, 0, 5, 5, 5, 4, 4, 5, 2, 2, 0, 0],
-       [0, 0, 0, 0, 0, 5, 4, 5, 4, 5, 2, 2, 0, 0],
-       [0, 0, 0, 0, 0, 0, 4, 4, 5, 5, 2, 2, 0, 0],
-       [0, 0, 0, 0, 0, 0, 0, 5, 5, 5, 2, 2, 0, 0],
-       [0, 0, 0, 0, 0, 0, 0, 0, 5, 5, 2, 2, 0, 0],
-       [0, 0, 0, 0, 0, 0, 0, 0, 0, 5, 2, 2, 0, 0],
-       [0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
-       [0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 7, 4, 4],
-       [0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 4, 4],
-       [0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 4],
-       [0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0]
-    ]
+      [0, 7, 7, 5, 5, 4, 5, 5, 4, 4, 5, 5, 2, 2],
+      [0, 0, 7, 5, 4, 5, 5, 4, 5, 4, 5, 5, 2, 2],
+      [0, 0, 0, 4, 5, 5, 4, 5, 5, 4, 5, 5, 2, 2],
+      [0, 0, 0, 0, 5, 5, 5, 4, 4, 5, 2, 2, 0, 0],
+      [0, 0, 0, 0, 0, 5, 4, 5, 4, 5, 2, 2, 0, 0],
+      [0, 0, 0, 0, 0, 0, 4, 4, 5, 5, 2, 2, 0, 0],
+      [0, 0, 0, 0, 0, 0, 0, 5, 5, 5, 2, 2, 0, 0],
+      [0, 0, 0, 0, 0, 0, 0, 0, 5, 5, 2, 2, 0, 0],
+      [0, 0, 0, 0, 0, 0, 0, 0, 0, 5, 2, 2, 0, 0],
+      [0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
+      [0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 7, 4, 4],
+      [0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 4, 4],
+      [0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 4],
+      [0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0]]
   end
 
   test "calculates link matrix (example 3)" do

diff --git a/test/rock/neighbours_test.exs b/test/rock/neighbours_test.exs
@@ -10,7 +10,7 @@ defmodule Rock.NeighboursTest do
     Point.new(["5", "6", "7"])
   ]
 
-  test "calculates neighbor matrix with jaccard coefficient" do
+  test "calculates neighbor matrix with jaccard coefficient (example 1)" do
     criterion = NeighbourCriterion.new(0.1)
 
     neighbor_matrix =
@@ -24,6 +24,46 @@ defmodule Rock.NeighboursTest do
     ]
   end
 
+  test "calculates neighbor matrix with jaccard coefficient (example 2)" do
+    criterion = NeighbourCriterion.new(0.5)
+    points = [
+      Point.new(["1", "2", "3"]),
+      Point.new(["1", "2", "4"]),
+      Point.new(["1", "2", "5"]),
+      Point.new(["1", "3", "4"]),
+      Point.new(["1", "3", "5"]),
+      Point.new(["1", "4", "5"]),
+      Point.new(["2", "3", "4"]),
+      Point.new(["2", "3", "5"]),
+      Point.new(["2", "4", "5"]),
+      Point.new(["3", "4", "5"]),
+      Point.new(["1", "2", "6"]),
+      Point.new(["1", "2", "7"]),
+      Point.new(["1", "6", "7"]),
+      Point.new(["2", "6", "7"])
+    ]
+    neighbor_matrix =
+      points
+      |> Neighbours.matrix(criterion)
+
+    [
+      [1, 1, 1, 1, 1, 0, 1, 1, 0, 0, 1, 1, 0, 0],
+      [1, 1, 1, 1, 0, 1, 1, 0, 1, 0, 1, 1, 0, 0],
+      [1, 1, 1, 0, 1, 1, 0, 1, 1, 0, 1, 1, 0, 0],
+      [1, 1, 0, 1, 1, 1, 1, 0, 0, 1, 0, 0, 0, 0],
+      [1, 0, 1, 1, 1, 1, 0, 1, 0, 1, 0, 0, 0, 0],
+      [0, 1, 1, 1, 1, 1, 0, 0, 1, 1, 0, 0, 0, 0],
+      [1, 1, 0, 1, 0, 0, 1, 1, 1, 1, 0, 0, 0, 0],
+      [1, 0, 1, 0, 1, 0, 1, 1, 1, 1, 0, 0, 0, 0],
+      [0, 1, 1, 0, 0, 1, 1, 1, 1, 1, 0, 0, 0, 0],
+      [0, 0, 0, 1, 1, 1, 1, 1, 1, 1, 0, 0, 0, 0],
+      [1, 1, 1, 0, 0, 0, 0, 0, 0, 0, 1, 1, 1, 1],
+      [1, 1, 1, 0, 0, 0, 0, 0, 0, 0, 1, 1, 1, 1],
+      [0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 1, 1, 1, 1],
+      [0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 1, 1, 1, 1]
+    ] = neighbor_matrix
+  end
+
   test "calculates neighbor matrix with custom similarity function" do
     similarity_function = fn(
         %Point{attributes: attributes1},