@@ -29,7 +29,8 @@ class PrefixspanSuite extends SparkFunSuite with MLlibTestSparkContext {
29
29
prefixSpanSeqs = read_baskets("prefixSpanSeqs", info = c("sequenceID","eventID","SIZE"))
30
30
freqItemSeq = cspade(
31
31
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 ))
33
34
resSeq = as(freqItemSeq, "data.frame")
34
35
resSeq
35
36
*/
@@ -64,23 +65,23 @@ class PrefixspanSuite extends SparkFunSuite with MLlibTestSparkContext {
64
65
val result1 = prefixspan.run(rdd)
65
66
val expectedValue1 = Array (
66
67
(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 ),
74
75
(Array (2 ), 2L ),
75
- (Array (2 ,1 ), 2L ),
76
+ (Array (2 , 1 ), 2L ),
76
77
(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 ),
82
83
(Array (4 ), 4L ),
83
- (Array (4 ,5 ), 2L ),
84
+ (Array (4 , 5 ), 2L ),
84
85
(Array (5 ), 3L )
85
86
)
86
87
assert(compareResult(expectedValue1, result1.collect()))
@@ -90,7 +91,7 @@ class PrefixspanSuite extends SparkFunSuite with MLlibTestSparkContext {
90
91
val expectedValue2 = Array (
91
92
(Array (1 ), 4L ),
92
93
(Array (3 ), 5L ),
93
- (Array (3 ,4 ), 3L ),
94
+ (Array (3 , 4 ), 3L ),
94
95
(Array (4 ), 4L ),
95
96
(Array (5 ), 3L )
96
97
)
@@ -100,18 +101,18 @@ class PrefixspanSuite extends SparkFunSuite with MLlibTestSparkContext {
100
101
val result3 = prefixspan.run(rdd)
101
102
val expectedValue3 = Array (
102
103
(Array (1 ), 4L ),
103
- (Array (1 ,3 ), 2L ),
104
+ (Array (1 , 3 ), 2L ),
104
105
(Array (1 ,4 ), 2L ),
105
- (Array (1 ,5 ), 2L ),
106
- (Array (2 ,1 ), 2L ),
106
+ (Array (1 , 5 ), 2L ),
107
+ (Array (2 , 1 ), 2L ),
107
108
(Array (2 ), 2L ),
108
109
(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 ),
113
114
(Array (4 ), 4L ),
114
- (Array (4 ,5 ), 2L ),
115
+ (Array (4 , 5 ), 2L ),
115
116
(Array (5 ), 3L )
116
117
)
117
118
assert(compareResult(expectedValue3, result3.collect()))
0 commit comments