@@ -4,7 +4,7 @@ import org.scalatest.{FlatSpec, Matchers}
44import org .sharpsw .leetcode .PartitionList ._
55
66class PartitionListSpec extends FlatSpec with Matchers {
7- " Test 001 " should " OK " in {
7+ " 1->4->3->2->5->2 and x = 2 " should " return 1-2-2-4-3-5 " in {
88 val node1 = new ListNode (1 )
99 val node2 = new ListNode (4 )
1010 val node3 = new ListNode (3 )
@@ -19,4 +19,36 @@ class PartitionListSpec extends FlatSpec with Matchers {
1919 val result = partition(node1, 3 )
2020 ListNodeUtils .mkString(result) shouldEqual " 1-2-2-4-3-5"
2121 }
22+
23+ " 1->4->3->2->5->2->3" should " return 1-2-2-4-3-5-3" in {
24+ val node1 = new ListNode (1 )
25+ val node2 = new ListNode (4 )
26+ val node3 = new ListNode (3 )
27+ val node4 = new ListNode (2 )
28+ val node5 = new ListNode (5 )
29+ val node6 = new ListNode (2 )
30+ val node7 = new ListNode (3 )
31+ node1.next = node2
32+ node2.next = node3
33+ node3.next = node4
34+ node4.next = node5
35+ node5.next = node6
36+ node6.next = node7
37+
38+ val result = partition(node1, 3 )
39+ ListNodeUtils .mkString(result) shouldEqual " 1-2-2-4-3-5-3"
40+ }
41+
42+ " 1->4->2-10" should " return 1-2-4-10" in {
43+ val node1 = new ListNode (1 )
44+ val node2 = new ListNode (4 )
45+ val node3 = new ListNode (2 )
46+ val node4 = new ListNode (10 )
47+ node1.next = node2
48+ node2.next = node3
49+ node3.next = node4
50+
51+ val result = partition(node1, 3 )
52+ ListNodeUtils .mkString(result) shouldEqual " 1-2-4-10"
53+ }
2254}
0 commit comments