1313import static fj .P .p ;
1414import static fj .data .Option .none ;
1515import static fj .data .Option .some ;
16+ import static fj .data .TreeMap .iterableTreeMap ;
1617import static org .hamcrest .CoreMatchers .equalTo ;
1718import static org .junit .Assert .assertEquals ;
1819import static org .junit .Assert .assertThat ;
@@ -29,7 +30,7 @@ public void split() {
2930 int pivot = 4 ;
3031 int max = 5 ;
3132 List <Integer > l = List .range (1 , max + 1 );
32- TreeMap <Integer , String > m2 = TreeMap . treeMap (Ord .intOrd , l .zip (l .map (i -> i .toString ())));
33+ TreeMap <Integer , String > m2 = iterableTreeMap (Ord .intOrd , l .zip (l .map (i -> i .toString ())));
3334 P3 <Set <String >, Option <String >, Set <String >> p = m2 .split (Ord .stringOrd , pivot );
3435
3536 // print debug info
@@ -51,7 +52,7 @@ public void split() {
5152 }
5253
5354 private static Set <String > toSetString (List <Integer > list ) {
54- return Set .fromList (Ord .stringOrd , list .map (i -> i .toString ()));
55+ return Set .iterableSet (Ord .stringOrd , list .map (i -> i .toString ()));
5556 }
5657
5758 @ Test
@@ -60,14 +61,14 @@ public void splitLookup() {
6061 int pivot = 4 ;
6162 int max = 5 ;
6263 List <Integer > l = List .range (1 , max + 1 );
63- TreeMap <Integer , String > m2 = TreeMap . treeMap (Ord .intOrd , l .zip (l .map (i -> i .toString ())));
64+ TreeMap <Integer , String > m2 = iterableTreeMap (Ord .intOrd , l .zip (l .map (i -> i .toString ())));
6465 P3 <TreeMap <Integer , String >, Option <String >, TreeMap <Integer , String >> p3 = m2 .splitLookup (pivot );
6566
6667 // create expected output
6768 List <Integer > leftList = List .range (1 , pivot );
68- TreeMap <Integer , String > leftMap = TreeMap . treeMap (Ord .intOrd , leftList .zip (leftList .map (i -> i .toString ())));
69+ TreeMap <Integer , String > leftMap = iterableTreeMap (Ord .intOrd , leftList .zip (leftList .map (i -> i .toString ())));
6970 List <Integer > rightList = List .range (pivot + 1 , max + 1 );
70- TreeMap <Integer , String > rightMap = TreeMap . treeMap (Ord .intOrd , rightList .zip (rightList .map (i -> i .toString ())));
71+ TreeMap <Integer , String > rightMap = iterableTreeMap (Ord .intOrd , rightList .zip (rightList .map (i -> i .toString ())));
7172
7273 // debug info
7374 if (true ) {
@@ -86,7 +87,7 @@ public void splitLookup() {
8687 public void toMutableMap () {
8788 int max = 5 ;
8889 List <List <Integer >> l = List .range (1 , max + 1 ).map (n -> List .single (n ));
89- TreeMap <List <Integer >, String > m2 = TreeMap . treeMap (Ord .listOrd (Ord .intOrd ), l .zip (l .map (i -> i .toString ())));
90+ TreeMap <List <Integer >, String > m2 = iterableTreeMap (Ord .listOrd (Ord .intOrd ), l .zip (l .map (i -> i .toString ())));
9091 Map <List <Integer >, String > mm = m2 .toMutableMap ();
9192 assertEquals (m2 .keys (), List .iterableList (mm .keySet ()));
9293 }
0 commit comments