extra/sequences/extras/extras-tests.factor

   1 USING: arrays ascii io io.streams.string kernel make math
   2 math.vectors random sequences sequences.extras strings
   3 tools.test vectors ;
   4
   5 IN: sequences.extras.tests
   6
   7 { { "a" "b" "c" "d" "ab" "bc" "cd" "abc" "bcd" "abcd" } } [ "abcd" all-subseqs ] unit-test
   8
   9 { { "a" "ab" "abc" "abcd" "b" "bc" "bcd" "c" "cd" "d" } }
  10 [ [ "abcd" [ , ] each-subseq ] { } make ] unit-test
  11
  12 { B{ 115 } } [ 1 2 "asdf" B{ } subseq-as ] unit-test
  13
  14 { "" } [ "abc" "def" longest-subseq ] unit-test
  15 { "abcd" } [ "abcd" "abcde" longest-subseq ] unit-test
  16 { "foo" } [ "foo" "foobar" longest-subseq ] unit-test
  17 { "foo" } [ "foobar" "foo" longest-subseq ] unit-test
  18
  19 { "" "" } [ "" "" CHAR: ? pad-longest ] unit-test
  20 { "abc" "def" } [ "abc" "def" CHAR: ? pad-longest ] unit-test
  21 { "   " "abc" } [ "" "abc" CHAR: \s pad-longest ] unit-test
  22 { "abc" "   " } [ "abc" "" CHAR: \s pad-longest ] unit-test
  23 { "abc..." "foobar" } [ "abc" "foobar" CHAR: . pad-longest ] unit-test
  24
  25 { { 0 1 0 1 } } [
  26     { 0 0 0 0 } { 1 3 } over [ 1 + ] change-nths
  27 ] unit-test
  28
  29 { V{ f t f } } [
  30     { 1 2 3 } [ even? ] selector* [ each ] dip
  31 ] unit-test
  32
  33 { { 1 3 5 } } [ { 1 2 3 4 5 6 } [ nip even? ] filter-index ] unit-test
  34
  35 { V{ 1 3 5 } } [ { 1 2 3 4 5 6 } [ nip even? ] V{ } filter-index-as ] unit-test
  36
  37 { { 1 3 5 } } [ { 1 2 3 4 5 6 } even-indices ] unit-test
  38
  39 { { 2 4 6 } } [ { 1 2 3 4 5 6 } odd-indices ] unit-test
  40
  41 { "a b c d e" }
  42 [ "a      b  \t \n \r  c   d \n    e   " [ blank? ] " " compact ] unit-test
  43
  44 { " a b c d e " }
  45 [ " a      b  c   d    e   " [ blank? ] " " collapse ] unit-test
  46
  47 { { "hello," " " "world!" " " " " } }
  48 [ "hello, world!  " [ blank? ] slice-when [ >string ] map ] unit-test
  49
  50 { "hello" } [ "hello" 0 rotate ] unit-test
  51 { "llohe" } [ "hello" 2 rotate ] unit-test
  52 { "hello" } [ "hello" dup 0 rotate! ] unit-test
  53 { "lohel" } [ "hello" dup 3 rotate! ] unit-test
  54
  55 { { } } [ { } [ ] map-concat ] unit-test
  56 { V{ 0 0 1 0 1 2 } } [ 4 iota [ iota ] map-concat ] unit-test
  57 { "abc" } [ "abc" [ 1string ] map-concat ] unit-test
  58 { "abc" } [ { 97 98 99 } [ 1string ] map-concat ] unit-test
  59 { { 97 98 99 } } [ "abc" [ 1string ] { } map-concat-as ] unit-test
  60 { "baz" { "foobaz" "barbaz" } }
  61 [ "baz" { { "foo" } { "bar" } } [ [ over append ] map ] map-concat ] unit-test
  62
  63 { { } } [ { } [ ] [ even? ] map-filter ] unit-test
  64 { "bcde" } [ "abcd" [ 1 + ] [ drop t ] map-filter ] unit-test
  65 { { 0 4 16 36 64 } } [ 10 iota [ sq ] [ even? ] { } map-filter-as ] unit-test
  66
  67 { V{ 0 4 16 36 64 } } [ 10 iota [ even? ] [ sq ] filter-map ] unit-test
  68 { { 2 6 10 14 18 } } [ 10 iota [ odd? ] [ 2 * ] { } filter-map-as ] unit-test
  69
  70 { 8 } [ 3 iota dup [ 1 + * ] 2map-sum ] unit-test
  71 { 4 } [ "hello" "jello" [ = ] 2count ] unit-test
  72
  73 { { } } [ { } round-robin ] unit-test
  74 { "ADEBFC" } [ { "ABC" "D" "EF" } round-robin >string ] unit-test
  75
  76 { { } } [ "ABC" [ ] { } trim-as ] unit-test
  77 { "ABC" } [ { 32 65 66 67 32 } [ blank? ] "" trim-as ] unit-test
  78
  79 { t } [ "ABC" dup [ blank? ] ?trim [ identity-hashcode ] same? ] unit-test
  80 { "ABC" } [ " ABC " [ blank? ] ?trim ] unit-test
  81
  82 { t } [ "ABC" dup [ blank? ] ?trim-head [ identity-hashcode ] same? ] unit-test
  83 { t } [ "ABC" dup [ blank? ] ?trim-tail [ identity-hashcode ] same? ] unit-test
  84 { "ABC " } [ " ABC " [ blank? ] ?trim-head ] unit-test
  85 { " ABC" } [ " ABC " [ blank? ] ?trim-tail ] unit-test
  86
  87 { "" } [ "" "" "" unsurround ] unit-test
  88 { "" } [ "  " " " " " unsurround ] unit-test
  89 { "foo.com" } [ "http://foo.com" "http://" "/" unsurround ] unit-test
  90
  91 { t } [ { 1 3 5 7 } [ even? ] none? ] unit-test
  92 { f } [ { 1 2 3 4 } [ even? ] none? ] unit-test
  93 { t } [ { } [ even? ] none? ] unit-test
  94
  95 { f } [ { 1 2 3 4 } [ even? ] one? ] unit-test
  96 { t } [ { 1 2 3 } [ even? ] one? ] unit-test
  97 { f } [ { } [ even? ] one? ] unit-test
  98
  99 { { { 5 8 0 } { 6 9 1 } { 7 10 2 } } } [ { 5 6 7 } { 8 9 10 } [ 3array ] 2map-index ] unit-test
 100
 101 { { } } [ { } <evens> >array ] unit-test
 102 { { 0 2 } } [ 4 iota <evens> >array ] unit-test
 103 { { 0 2 4 } } [ 5 iota <evens> >array ] unit-test
 104
 105 { { } } [ { } <odds> >array ] unit-test
 106 { { 1 3 } } [ 5 iota <odds> >array ] unit-test
 107 { { 1 3 5 } } [ 6 iota <odds> >array ] unit-test
 108
 109 { 1 } [ { 1 7 3 7 6 3 7 } arg-max ] unit-test
 110 { 0 } [ { 1 7 3 7 6 3 7 } arg-min ] unit-test
 111 { V{ 0 4 } } [ { 5 3 2 10 5 } [ 5 = ] arg-where ] unit-test
 112 { { 2 1 0 4 3 } } [ { 5 3 2 10 5 } arg-sort ] unit-test
 113
 114 { t } [ { 1 2 3 4 5 } 1 first= ] unit-test
 115 { t } [ { 1 2 3 4 5 } 2 second= ] unit-test
 116 { t } [ { 1 2 3 4 5 } 3 third= ] unit-test
 117 { t } [ { 1 2 3 4 5 } 4 fourth= ] unit-test
 118 { t } [ { 1 2 3 4 5 } 5 last= ] unit-test
 119 { t } [ 4 { 1 2 3 4 5 } 5 nth= ] unit-test
 120
 121 { t } [ { 1 2 3 4 5 } [ 1 = ] first? ] unit-test
 122 { t } [ { 1 2 3 4 5 } [ 2 = ] second? ] unit-test
 123 { t } [ { 1 2 3 4 5 } [ 3 = ] third? ] unit-test
 124 { t } [ { 1 2 3 4 5 } [ 4 = ] fourth? ] unit-test
 125 { t } [ { 1 2 3 4 5 } [ 5 = ] last? ] unit-test
 126 { t } [ 4 { 1 2 3 4 5 } [ 5 = ] nth? ] unit-test
 127
 128 { { 97 115 100 102 } } [
 129     "asdf" [ [ read1 ] loop>array ] with-string-reader
 130 ] unit-test
 131
 132 { V{ 97 115 100 102 } } [
 133     "asdf" [ [ read1 ] V{ } loop>sequence ] with-string-reader
 134 ] unit-test
 135
 136 { "" } [ { } "" reverse-as ] unit-test
 137 { "ABC" } [ { 67 66 65 } "" reverse-as ] unit-test
 138
 139 { V{ 1 } } [ 1 0 V{ } [ insert-nth! ] keep ] unit-test
 140 { V{ 1 2 3 4 } } [ 2 1 V{ 1 3 4 } [ insert-nth! ] keep ] unit-test
 141
 142 { "abc" } [ B{ 97 98 99 100 101 102 103 } 3 "" head-as ] unit-test
 143 { "abcd" } [ B{ 97 98 99 100 101 102 103 } 3 "" head*-as ] unit-test
 144 { "defg" } [ B{ 97 98 99 100 101 102 103 } 3 "" tail-as ] unit-test
 145 { "efg" } [ B{ 97 98 99 100 101 102 103 } 3 "" tail*-as ] unit-test
 146
 147 { { 1 0 0 1 0 0 0 1 0 0 } }
 148 [ 1 { 0 3 7 } 10 0 <array> [ set-nths ] keep ] unit-test
 149
 150 { { 1 0 0 1 0 0 0 1 0 0 } }
 151 [ 1 { 0 3 7 } 10 0 <array> [ set-nths-unsafe ] keep ] unit-test
 152
 153 { V{ 1 } } [ 1 flatten1 ] unit-test
 154 { { 1 2 3 } } [ { 1 2 3 } flatten1 ] unit-test
 155 { { 1 2 3 { { 4 } } } } [ { 1 { 2 } { 3 { { 4 } } } } flatten1 ] unit-test
 156
 157 { t 3 3 } [ 10 iota [ [ odd? ] [ 1 > ] bi* and ] map-find-index ] unit-test
 158 { f f f } [ 10 iota [ [ odd? ] [ 9 > ] bi* and ] map-find-index ] unit-test
 159
 160 { "abcdef" } [ f f "abcdef" subseq* ] unit-test
 161 { "abcdef" } [ 0 f "abcdef" subseq* ] unit-test
 162 { "ab" } [ f 2 "abcdef" subseq* ] unit-test
 163 { "cdef" } [ 2 f "abcdef" subseq* ] unit-test
 164 { "cd" } [ -4 -2 "abcdef" subseq* ] unit-test
 165
 166 { "foo" "" } [ "foo" [ blank? ] cut-when ] unit-test
 167 { "foo" " " } [ "foo " [ blank? ] cut-when ] unit-test
 168 { "" " foo" } [ " foo" [ blank? ] cut-when ] unit-test
 169 { "foo" " bar" } [ "foo bar" [ blank? ] cut-when ] unit-test
 170
 171 { { 4 0 3 1 2 } } [ { 0 4 1 3 2 } 5 iota [ nth* ] curry map ] unit-test
 172
 173 { 1 "beef" } [ { "chicken" "beef" "moose" } [ length ] infimum-by* ] unit-test
 174 { 0 "chicken" } [ { "chicken" "beef" "moose" } [ length ] supremum-by* ] unit-test
 175 { 2 "moose" } [ { "chicken" "beef" "moose" } [ first ] supremum-by* ] unit-test
 176
 177 { 3/10 } [ 10 iota [ 3 < ] count* ] unit-test
 178
 179 { { 0 } } [ "ABA" "ABABA" start-all ] unit-test
 180 { { 0 2 } } [ "ABA" "ABABA" start-all* ] unit-test
 181 { { 0 3 } } [ "ABA" "ABAABA" start-all ] unit-test
 182 { 1 } [ "ABA" "ABABA" count-subseq ] unit-test
 183 { 2 } [ "ABA" "ABABA" count-subseq* ] unit-test
 184
 185 { 120000 } [ { 10 20 30 40 50 60 } 1 [ * ] 3 reduce-from ] unit-test