80 lines
1.8 KiB
Hy
80 lines
1.8 KiB
Hy
(import [functools [wraps]])
|
|
|
|
|
|
(defn test-sharp-macro []
|
|
"Test a basic sharp macro"
|
|
(defsharp ^ [expr]
|
|
expr)
|
|
|
|
(assert (= #^"works" "works")))
|
|
|
|
|
|
(defn test-sharp-macro-expr []
|
|
"Test basic exprs like lists and arrays"
|
|
(defsharp n [expr]
|
|
(get expr 1))
|
|
|
|
(assert (= #n[1 2] 2))
|
|
(assert (= #n(1 2) 2)))
|
|
|
|
|
|
(defn test-sharp-macro-override []
|
|
"Test if we can override function symbols"
|
|
(defsharp + [n]
|
|
(+ n 1))
|
|
|
|
(assert (= #+2 3)))
|
|
|
|
|
|
(defn test-sharp-macros-macros []
|
|
"Test if defsharp is actually a macro"
|
|
(defsharp t [expr]
|
|
`(, ~@expr))
|
|
|
|
(def a #t[1 2 3])
|
|
|
|
(assert (= (type a) tuple))
|
|
(assert (= (, 1 2 3) a)))
|
|
|
|
|
|
(defn test-sharp-macro-string-name []
|
|
"Test if defsharp accepts a string as a macro name."
|
|
|
|
(defsharp "." [expr]
|
|
expr)
|
|
|
|
(assert (= #."works" "works")))
|
|
|
|
|
|
(defn test-builtin-decorator-sharp []
|
|
(defn increment-arguments [func]
|
|
"Increments each argument passed to the decorated function."
|
|
((wraps func)
|
|
(fn [&rest args &kwargs kwargs]
|
|
(apply func
|
|
(map inc args)
|
|
(dict-comp k (inc v) [[k v] (.items kwargs)])))))
|
|
|
|
#@(increment-arguments
|
|
(defn foo [&rest args &kwargs kwargs]
|
|
"Bar."
|
|
(, args kwargs)))
|
|
|
|
;; The decorator did what it was supposed to
|
|
(assert (= (, (, 2 3 4) {"quux" 5 "baz" 6})
|
|
(foo 1 2 3 :quux 4 :baz 5)))
|
|
|
|
;; @wraps preserved the docstring and __name__
|
|
(assert (= "foo" (. foo --name--)))
|
|
(assert (= "Bar." (. foo --doc--)))
|
|
|
|
;; We can use the #@ sharp macro to apply more than one decorator
|
|
#@(increment-arguments
|
|
increment-arguments
|
|
(defn double-foo [&rest args &kwargs kwargs]
|
|
"Bar."
|
|
(, args kwargs)))
|
|
|
|
(assert (= (, (, 3 4 5) {"quux" 6 "baz" 7})
|
|
(double-foo 1 2 3 :quux 4 :baz 5))))
|