extra/assocs/extras/extras-tests.factor

   1 USING: assocs.extras kernel math sequences tools.test ;
   2
   3 { f } [ f { } deep-at ] unit-test
   4 { f } [ f { "foo" } deep-at ] unit-test
   5 { f } [ H{ } { 1 2 3 } deep-at ] unit-test
   6 { f } [ H{ { "a" H{ { "b" 1 } } } } { "a" "c" } deep-at ] unit-test
   7 { 1 } [ H{ { "a" H{ { "b" 1 } } } } { "a" "b" } deep-at ] unit-test
   8 { 4 } [ H{ { 1 H{ { 2 H{ { 3 4 } } } } } } { 1 2 3 } deep-at ] unit-test
   9
  10 { H{ { 2 1 } { 4 3 } } } [ H{ { 1 2 } { 3 4 } } assoc-invert ] unit-test
  11
  12 { H{ { "a" V{ 2 5 } } { "b" V{ 3 } } { "c" V{ 10 } } } }
  13 [
  14     { H{ { "a" 2 } { "b" 3 } } H{ { "a" 5 } { "c" 10 } } }
  15     [ ] [ assoc-merge ] map-reduce
  16 ] unit-test
  17
  18 { H{ } } [ H{ { 1 2 } } 2 over delete-value-at ] unit-test
  19 { H{ { 1 2 } } } [ H{ { 1 2 } } 3 over delete-value-at ] unit-test
  20
  21 {
  22     H{ { 1 3 } { 2 3 } }
  23 } [
  24     {
  25         { { 1 2 } 3 }
  26     } expand-keys-set-at
  27 ] unit-test
  28
  29 {
  30     H{ { 3 4 } }
  31 } [
  32     {
  33         { 3 { 1 2 } } { 3 4 }
  34     } expand-values-set-at
  35 ] unit-test
  36
  37 {
  38     H{ { 1 V{ 3 } } { 2 V{ 3 } } }
  39 } [
  40     {
  41         { { 1 2 } 3 }
  42     } expand-keys-push-at
  43 ] unit-test
  44
  45 {
  46     H{ { 3 V{ 1 2 4 } } }
  47 } [
  48     {
  49         { 3 { 1 2 } } { 3 4 }
  50     } expand-values-push-at
  51 ] unit-test
  52
  53 {
  54     H{ { 1 [ sq ] } { 2 [ sq ] } }
  55 } [
  56     { { { 1 2 { 1 } { 2 } { 1 1 } } [ sq ] } } flatten-keys
  57 ] unit-test