From 83b704afb9d67fa7838ef0540f24a6aaf0077439 Mon Sep 17 00:00:00 2001 From: Alex Rubinsteyn Date: Mon, 25 Mar 2024 22:54:56 -0400 Subject: [PATCH] added common.py --- tests/common.py | 59 +++++++++++++++++++++++++++++++++++++++++++++++++ 1 file changed, 59 insertions(+) create mode 100644 tests/common.py diff --git a/tests/common.py b/tests/common.py new file mode 100644 index 0000000..3dae4b2 --- /dev/null +++ b/tests/common.py @@ -0,0 +1,59 @@ +# Licensed under the Apache License, Version 2.0 (the "License"); +# you may not use this file except in compliance with the License. +# You may obtain a copy of the License at +# +# http://www.apache.org/licenses/LICENSE-2.0 +# +# Unless required by applicable law or agreed to in writing, software +# distributed under the License is distributed on an "AS IS" BASIS, +# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. +# See the License for the specific language governing permissions and +# limitations under the License. + +def ok_(a, s=None): + if s is None: + assert a + else: + assert a, s + +def eq_(a, b, s=None): + if s is None: + assert a == b + else: + assert a == b, s + +def neq_(a, b, s=None): + if s is None: + assert a != b + else: + assert a != b, s + +def gt_(a, b, s=None): + if s is None: + assert a > b + else: + assert a > b, s + +def lt_(a, b, s=None): + if s is None: + assert a < b + else: + assert a < b, s + +def gte_(a, b, s=None): + if s is None: + assert a >= b + else: + assert a >= b, s + +def lte_(a, b, s=None): + if s is None: + assert a <= b + else: + assert a <= b, s + +def almost_eq_(a, b, tol=1e-6, s=None): + if s is None: + assert abs(a - b) < tol + else: + assert abs(a - b) < tol, s \ No newline at end of file