assocs.extras: add intersect-keys-as and unit tests
authorDoug Coleman <doug.coleman@gmail.com>
Sat, 20 Aug 2022 21:50:04 +0000 (17:50 -0400)
committerDoug Coleman <doug.coleman@gmail.com>
Sat, 20 Aug 2022 21:50:37 +0000 (17:50 -0400)
extra/assocs/extras/extras-tests.factor
extra/assocs/extras/extras.factor

index c8233c6d251588d0f133cd7bbf8b7646b99b6e29..cf7ce6ce37a80e4b699d75f9fbae6bd19316a2af 100644 (file)
@@ -296,3 +296,12 @@ USING: arrays assocs.extras kernel math math.order sequences tools.test ;
     [ [ dup 1 + 2array ] dip ] collect-assoc-by-multi
 ] unit-test
 
+{ H{ { 1 2 } { 3 4 } } } [
+    H{ { 1 2 } { 3 4 } { 5 6 } }
+    { 1 3 } intersect-keys
+] unit-test
+
+{ { { 1 2 } { 3 4 } } } [
+    H{ { 1 2 } { 3 4 } { 5 6 } }
+    { 1 3 } { } intersect-keys-as
+] unit-test
index 8aee7d6d9db110e65d14d15979093f9d2586ae93..7073a074382ca3ed6fa105dcb0466d9b8327c444 100644 (file)
@@ -217,8 +217,11 @@ PRIVATE>
 : flatten-values ( assoc -- assoc' )
     dup any-multi-value? [ expand-values-set-at flatten-values ] when ;
 
+: intersect-keys-as ( assoc seq exemplar -- elts )
+  [ [ of ] with ] dip zip-with-as sift-values ; inline
+
 : intersect-keys ( assoc seq -- elts )
-    [ of ] with zip-with sift-values ; inline
+    over intersect-keys-as ; inline
 
 : values-of ( assoc seq -- seq' )
     [ of ] with map ; inline