@@ -29,7 +29,8 @@ class PrefixspanSuite extends SparkFunSuite with MLlibTestSparkContext {
2929 prefixSpanSeqs = read_baskets("prefixSpanSeqs", info = c("sequenceID","eventID","SIZE"))
3030 freqItemSeq = cspade(
3131 prefixSpanSeqs,
32- parameter = list(support = 2 / length(unique(transactionInfo(prefixSpanSeqs)$sequenceID)), maxlen = 2 ))
32+ parameter = list(support =
33+ 2 / length(unique(transactionInfo(prefixSpanSeqs)$sequenceID)), maxlen = 2 ))
3334 resSeq = as(freqItemSeq, "data.frame")
3435 resSeq
3536 */
@@ -64,23 +65,23 @@ class PrefixspanSuite extends SparkFunSuite with MLlibTestSparkContext {
6465 val result1 = prefixspan.run(rdd)
6566 val expectedValue1 = Array (
6667 (Array (1 ), 4L ),
67- (Array (1 ,3 ),2L ),
68- (Array (1 ,3 , 4 ), 2L ),
69- (Array (1 ,3 , 4 , 5 ), 2L ),
70- (Array (1 ,3 , 5 ), 2L ),
71- (Array (1 ,4 ), 2L ),
72- (Array (1 ,4 , 5 ), 2L ),
73- (Array (1 ,5 ), 2L ),
68+ (Array (1 , 3 ), 2L ),
69+ (Array (1 , 3 , 4 ), 2L ),
70+ (Array (1 , 3 , 4 , 5 ), 2L ),
71+ (Array (1 , 3 , 5 ), 2L ),
72+ (Array (1 , 4 ), 2L ),
73+ (Array (1 , 4 , 5 ), 2L ),
74+ (Array (1 , 5 ), 2L ),
7475 (Array (2 ), 2L ),
75- (Array (2 ,1 ), 2L ),
76+ (Array (2 , 1 ), 2L ),
7677 (Array (3 ), 5L ),
77- (Array (3 ,1 ), 2L ),
78- (Array (3 ,3 ), 2L ),
79- (Array (3 ,4 ), 3L ),
80- (Array (3 ,4 , 5 ), 2L ),
81- (Array (3 ,5 ), 2L ),
78+ (Array (3 , 1 ), 2L ),
79+ (Array (3 , 3 ), 2L ),
80+ (Array (3 , 4 ), 3L ),
81+ (Array (3 , 4 , 5 ), 2L ),
82+ (Array (3 , 5 ), 2L ),
8283 (Array (4 ), 4L ),
83- (Array (4 ,5 ), 2L ),
84+ (Array (4 , 5 ), 2L ),
8485 (Array (5 ), 3L )
8586 )
8687 assert(compareResult(expectedValue1, result1.collect()))
@@ -90,7 +91,7 @@ class PrefixspanSuite extends SparkFunSuite with MLlibTestSparkContext {
9091 val expectedValue2 = Array (
9192 (Array (1 ), 4L ),
9293 (Array (3 ), 5L ),
93- (Array (3 ,4 ), 3L ),
94+ (Array (3 , 4 ), 3L ),
9495 (Array (4 ), 4L ),
9596 (Array (5 ), 3L )
9697 )
@@ -100,18 +101,18 @@ class PrefixspanSuite extends SparkFunSuite with MLlibTestSparkContext {
100101 val result3 = prefixspan.run(rdd)
101102 val expectedValue3 = Array (
102103 (Array (1 ), 4L ),
103- (Array (1 ,3 ), 2L ),
104+ (Array (1 , 3 ), 2L ),
104105 (Array (1 ,4 ), 2L ),
105- (Array (1 ,5 ), 2L ),
106- (Array (2 ,1 ), 2L ),
106+ (Array (1 , 5 ), 2L ),
107+ (Array (2 , 1 ), 2L ),
107108 (Array (2 ), 2L ),
108109 (Array (3 ), 5L ),
109- (Array (3 ,1 ), 2L ),
110- (Array (3 ,3 ), 2L ),
111- (Array (3 ,4 ), 3L ),
112- (Array (3 ,5 ), 2L ),
110+ (Array (3 , 1 ), 2L ),
111+ (Array (3 , 3 ), 2L ),
112+ (Array (3 , 4 ), 3L ),
113+ (Array (3 , 5 ), 2L ),
113114 (Array (4 ), 4L ),
114- (Array (4 ,5 ), 2L ),
115+ (Array (4 , 5 ), 2L ),
115116 (Array (5 ), 3L )
116117 )
117118 assert(compareResult(expectedValue3, result3.collect()))
0 commit comments