mdebray/tvl-depot
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

feat(wpcarro/emacs): Support list-chunk function

Browse source 
Would be nice to remove the top-level `if` statement, but I can't be bothered to
debug the first unit test without that workaround.

Change-Id: I4ba576bda915eeb11f9fbfeb5d95d8e5668fd0bd
Reviewed-on: https://cl.tvl.fyi/c/depot/+/6022
Reviewed-by: wpcarro <wpcarro@gmail.com>
Autosubmit: wpcarro <wpcarro@gmail.com>
Tested-by: BuildkiteCI
This commit is contained in:
William Carroll 2022-08-01 13:43:34 -07:00 committed by clbot
parent 852cc4a9c8
commit 5417b512e7
2 changed files with 28 additions and 0 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

18
users/wpcarro/emacs/pkgs/list/list.el
View file

@ -160,6 +160,24 @@
(list-cons x acc)))
xs)))
(defun list-chunk (n xs)
"Chunk XS into lists of size N."
(if (> n (length xs))
(list xs)
(->> xs
(list-reduce '(:curr () :result ())
(lambda (x acc)
(let ((curr (plist-get acc :curr))
(result (plist-get acc :result)))
(if (= (- n 1) (length curr))
`(:curr () :result ,(list-cons (list-reverse (list-cons x curr)) result))
`(:curr ,(list-cons x curr) :result ,result)))))
(funcall (lambda (xs)
(let ((curr (plist-get xs :curr))
(result (plist-get xs :result)))
(if curr (list-cons curr result)) result)))
list-reverse)))
;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;
;; Predicates
;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;

10
users/wpcarro/emacs/pkgs/list/tests.el
View file

@ -42,3 +42,13 @@
(ert-deftest list-join ()
(should (equal "foo-bar-baz"
(list-join "-" '("foo" "bar" "baz")))))
(ert-deftest list-chunk ()
(should (equal '((1 2 3 4 5 6))
(list-chunk 7 '(1 2 3 4 5 6))))
(should (equal '((1) (2) (3) (4) (5) (6))
(list-chunk 1 '(1 2 3 4 5 6))))
(should (equal '((1 2 3) (4 5 6))
(list-chunk 3 '(1 2 3 4 5 6))))
(should (equal '((1 2) (3 4) (5 6))
(list-chunk 2 '(1 2 3 4 5 6)))))