summaryrefslogtreecommitdiffstats
path: root/lib/ansible/plugins/test/mathstuff.py
diff options
context:
space:
mode:
Diffstat (limited to 'lib/ansible/plugins/test/mathstuff.py')
-rw-r--r--lib/ansible/plugins/test/mathstuff.py62
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,
+ }