wassname/catalyst
1
0
Fork 0
mirror of https://github.com/wassname/catalyst.git synced 2026-07-15 11:22:18 +08:00
Code Issues Packages Projects Releases Wiki Activity

TEST: More thorough testing of boolean binops.

Browse Source
This commit is contained in:
Scott Sanderson
2016-02-10 18:45:00 -05:00
parent e7269a880c
commit dfb1832231
1 changed files with 15 additions and 9 deletions
Download Patch File Download Diff File Expand all files Collapse all files
tests/pipeline/test_numerical_expression.py
+15 -9
View File
@@ -1,3 +1,4 @@
from itertools import permutations
from operator import (
add,
ge,
@@ -498,15 +499,20 @@ class NumericalExpressionTestCase(TestCase):
self.check_output(eye_filter, eye_mask)
self.check_output(first_row_filter, first_row_mask)
def gen_boolops(a, b, c):
yield (a & b) & c
yield (a & b) | c
yield (a | b) & c
yield (a | b) | c
yield a & (b & c)
yield a & (b | c)
yield a | (b & c)
yield a | (b | c)
def gen_boolops(x, y, z):
"""
Generate all possible interleavings of & and | between all possible
orderings of x, y, and z.
"""
for a, b, c in permutations([x, y, z]):
yield (a & b) & c
yield (a & b) | c
yield (a | b) & c
yield (a | b) | c
yield a & (b & c)
yield a & (b | c)
yield a | (b & c)
yield a | (b | c)
exprs = gen_boolops(eye_filter, custom_filter, first_row_filter)
arrays = gen_boolops(eye_mask, custom_filter_mask, first_row_mask)