assocs.extra: Add a word to keep only certain keys in an assoc to the same assoc...

[factor.git] / extra / assocs / extras / extras-tests.factor

diff --git a/extra/assocs/extras/extras-tests.factor b/extra/assocs/extras/extras-tests.factor
index 3034a01784382694f81252b7c4b0e1f55a64c4c7..23dd663aff2cf6793a9c34644478ad641fa21ac5 100644 (file)
--- a/extra/assocs/extras/extras-tests.factor
+++ b/extra/assocs/extras/extras-tests.factor
@@ -54,4 +54,30 @@ USING: assocs.extras kernel math sequences tools.test ;
     H{ { 1 [ sq ] } { 2 [ sq ] } }
 } [
     { { { 1 2 { 1 } { 2 } { 1 1 } } [ sq ] } } flatten-keys
+] unit-test
+
+{
+    H{ { "1" 1 } { "2" 2 } }
+} [
+    H{ { "1" 1 } { "2" 2 } { "3" 3 } }
+    { "1" "2" "2" }
+    rekey-new-assoc
+] unit-test
+
+{ f } [
+    H{ { "1" 1 } { "2" 2 } { "3" 3 } }
+    [ { "1" "2" "2" } rekey-new-assoc ] keep eq?
+] unit-test
+
+{
+    H{ { "1" 1 } { "2" 2 } }
+} [
+    H{ { "1" 1 } { "2" 2 } { "3" 3 } }
+    { "1" "2" "2" }
+    rekey-assoc
+] unit-test
+
+{ t } [
+    H{ { "1" 1 } { "2" 2 } { "3" 3 } }
+    [ { "1" "2" "2" } rekey-assoc ] keep eq?
 ] unit-test
\ No newline at end of file