hy/tests/native_tests/with_test.hy
2019-02-07 08:57:35 -05:00

71 lines
1.7 KiB
Hy

;; Copyright 2019 the authors.
;; This file is part of Hy, which is free software licensed under the Expat
;; license. See the LICENSE.
(defclass WithTest [object]
(defn --init-- [self val]
(setv self.val val)
None)
(defn --enter-- [self]
self.val)
(defn --exit-- [self type value traceback]
(setv self.val None)))
(defn test-single-with []
"NATIVE: test a single with"
(with [t (WithTest 1)]
(assert (= t 1))))
(defn test-two-with []
"NATIVE: test two withs"
(with [t1 (WithTest 1)
t2 (WithTest 2)]
(assert (= t1 1))
(assert (= t2 2))))
(defn test-thrice-with []
"NATIVE: test three withs"
(with [t1 (WithTest 1)
t2 (WithTest 2)
t3 (WithTest 3)]
(assert (= t1 1))
(assert (= t2 2))
(assert (= t3 3))))
(defn test-quince-with []
"NATIVE: test four withs, one with no args"
(with [t1 (WithTest 1)
t2 (WithTest 2)
t3 (WithTest 3)
_ (WithTest 4)]
(assert (= t1 1))
(assert (= t2 2))
(assert (= t3 3))))
(defclass SuppressZDE [object]
(defn --enter-- [self])
(defn --exit-- [self exc-type exc-value traceback]
(and (not (none? exc-type)) (issubclass exc-type ZeroDivisionError))))
(defn test-exception-suppressing-with []
; https://github.com/hylang/hy/issues/1320
(setv x (with [(SuppressZDE)] 5))
(assert (= x 5))
(setv y (with [(SuppressZDE)] (/ 1 0)))
(assert (none? y))
(setv z (with [(SuppressZDE)] (/ 1 0) 5))
(assert (none? z))
(defn f [] (with [(SuppressZDE)] (/ 1 0)))
(assert (none? (f)))
(setv w 7 l [])
(setv w (with [(SuppressZDE)] (.append l w) (/ 1 0) 5))
(assert (none? w))
(assert (= l [7])))