diff options
Diffstat (limited to 'lib/ansible/plugins/test/mathstuff.py')
-rw-r--r-- | lib/ansible/plugins/test/mathstuff.py | 62 |
1 files changed, 62 insertions, 0 deletions
diff --git a/lib/ansible/plugins/test/mathstuff.py b/lib/ansible/plugins/test/mathstuff.py new file mode 100644 index 0000000..9a3f467 --- /dev/null +++ b/lib/ansible/plugins/test/mathstuff.py @@ -0,0 +1,62 @@ +# (c) 2016, Ansible, Inc +# +# This file is part of Ansible +# +# Ansible is free software: you can redistribute it and/or modify +# it under the terms of the GNU General Public License as published by +# the Free Software Foundation, either version 3 of the License, or +# (at your option) any later version. +# +# Ansible is distributed in the hope that it will be useful, +# but WITHOUT ANY WARRANTY; without even the implied warranty of +# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the +# GNU General Public License for more details. +# +# You should have received a copy of the GNU General Public License +# along with Ansible. If not, see <http://www.gnu.org/licenses/>. + +from __future__ import (absolute_import, division, print_function) +__metaclass__ = type + +import math + + +def issubset(a, b): + return set(a) <= set(b) + + +def issuperset(a, b): + return set(a) >= set(b) + + +def isnotanumber(x): + try: + return math.isnan(x) + except TypeError: + return False + + +def contains(seq, value): + '''Opposite of the ``in`` test, allowing use as a test in filters like ``selectattr`` + + .. versionadded:: 2.8 + ''' + return value in seq + + +class TestModule: + ''' Ansible math jinja2 tests ''' + + def tests(self): + return { + # set theory + 'subset': issubset, + 'issubset': issubset, + 'superset': issuperset, + 'issuperset': issuperset, + 'contains': contains, + + # numbers + 'nan': isnotanumber, + 'isnan': isnotanumber, + } |